码迷,mamicode.com
首页 > 编程语言 > 详细

poj3468区间加减查找——树状数组区间修改查询

时间:2018-02-11 00:12:03      阅读:258      评论:0      收藏:0      [点我收藏+]

标签:blog   代码   http   公式   can   namespace   blank   +=   树状数组   

题目:http://poj.org/problem?id=3468

增加一个更改量数组,施以差值用法则区间修改变为单位置修改;

利用公式可通过树状数组维护两个数组:f与g而直接求出区间和。

代码如下:

#include<iostream>
#include<cstdio>
using namespace std;
long long n,q,a[100005]/*,b[100005],bg[100005]*/,l,r,ad;
long long f[100005],g[100005],sum[100005];//f[x]为更改量 
char dc;
long long query(long long x)
{
	long long k=x;
	long long summ=0;
	for(;x;x-=x&-x)
		summ+=f[x];
	long long s2=0;
	x=k;
	for(;x;x-=x&-x)
		s2+=g[x];
	return (k+1)*summ-s2;
}
void add(long long x,long long y)//x单位置增加y
{
	long long k=x;
	for(x;x<=n;x+=x&-x)
		f[x]+=y,g[x]+=k*y;
}
int main()
{
	scanf("%lld%lld",&n,&q);
	for(long long i=1;i<=n;i++)
	{
		scanf("%lld",&sum[i]);
		sum[i]+=sum[i-1];
//		b[i]=a[i]-a[i-1];
//		for(int j=i-(i&-i)+1;j<=i;j++)f[i]+=b[j];//差值 
//		bg[i]=i*b[i];
//		for(int j=i-(i&-i)+1;j<=i;j++)g[i]+=bg[j];
//		g[i]=i*f[i];
	}
	for(long long i=1;i<=q;i++)
	{
		scanf(" %c",&dc);
		if(dc==‘C‘)
		{
			scanf("%lld%lld%lld",&l,&r,&ad);
			add(l,ad);add(r+1,-ad);
		}
		if(dc==‘Q‘)
		{
			scanf("%lld%lld",&l,&r);
			long long ans=query(r)+sum[r]-query(l-1)-sum[l-1];
			printf("%lld\n",ans);
		}
	}
	return 0;
}

  

poj3468区间加减查找——树状数组区间修改查询

标签:blog   代码   http   公式   can   namespace   blank   +=   树状数组   

原文地址:https://www.cnblogs.com/Zinn/p/8440045.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!