树状数组 区间修改区间查询

树状数组 区间修改+区间查询

树状数组的区间修改和区间查询需要维护 两个 树状数组

首先还是先利用区间修改 单点查询用过的差分思想（将区间的修改操作变成单点的修改），我们先搞一个c数组作为差分数组

  a数组是原数组，c数组是差分数组。

简单可证：

  

即 a[1]=c[1],  a[2]=c[2]+c[1],  a[3]=c[3]+c[2]+c[1];

那么可以得到以下式子:

a[1]+a[2]+...+a[i]

=c[1]+(c[1]+c[2])+...+(c[1]+c[2]+...+c[i])

=i*c[1]+(i-1)*c[2]+...+c[i]

=i*(c[1]+c[2]+...+c[i])-1*c[2]-...-(i-1)*c[i]

于是，我们再搞一个数组c1

那之前的式子就可以表示为

 

然后再搞两个树状数组来分别维护一下c数组和c1数组的区间和就ok了

下面做一个类似的题目：codevs 1082的AC代码

本题是一个裸题，非常的简单，读者可自行阅读理解，学会使用树状数组区间修改区间查询的方法

#include<bits/stdc++.h>
using namespace std;
long long n,m,a[200010],b[200010],c[200010],bb[200010],c1[200010];
long long lowbit(long long x)
{
	return x&-x;
}
void modify(int x,long long y)
{
	for (int i=x;i<=n;i+=lowbit(i))
		b[i]+=y;
}
long long query(long long x)
{
	long long ans=0;
	for (int i=x;i>0;i-=lowbit(i))
		ans+=b[i];
	return ans;
}
void modifyy(long x,long long y)
{
	for (int i=x;i<=n;i+=lowbit(i))
		bb[i]+=y;
}
long long queryy(long long x)
{
	long long ans=0;
	for (int i=x;i>0;i-=lowbit(i))
		ans+=bb[i];
	return ans;
}
int main()
{
	cin>>n;
	for (int i=1;i<=n;i++)
	{
		cin>>a[i];
		c[i]=a[i]-a[i-1];
		c1[i]=(i-1)*c[i];
	}
	for (int i=1;i<=n;i++)
		modify(i,c[i]);
	for (int i=1;i<=n;i++)
		modifyy(i,c1[i]);
	cin>>m;
	for (int i=1;i<=m;i++)
	{
		long long p;
		cin>>p;
		if (p==1)
		{
			long long xx,yy,zz;
			cin>>xx>>yy>>zz;
			modify(xx,zz);
			modify(yy+1,-zz);
			modifyy(xx,(xx-1)*zz);
			modifyy(yy+1,-yy*zz);
		}
		if (p==2)
		{
			long long xx,yy,ss;
			cin>>xx>>yy;
			ss=(yy*query(yy)-queryy(yy))-((xx-1)*query(xx-1)-queryy(xx-1));
			cout<<ss<<endl;
		}
	}
	return 0;
}