【Ybtoj 第14章例4】区间修改区间查询【树状数组】

---

解题思路

考虑用 差分数组，将$[x,y]$每一个数都加k，设在$i$位置上的差分数组是$d_i$，将$d_x+=k,d_{y+1}-=k$。那么此时$a_i$就是$[1,i]$之间数的累加和，考虑维护前缀和，则
${\sum }_{i=1}^{n}a[i]$ = ${\sum }_{i=1}^n$${\sum }_{j=1}^{i}d[i]$

我们会发现，其中$d_1$出现了$n$次，$d_2$出现了$n-1$次，$d_i$出现了$n-i+1$次，所以有
${\sum }_{i=1}^{n}a[i]$ = ${\sum }_{i=1}^n（d_i\ast (n+1）-d_i\ast i)$

树状数组要维护两种数：一种是$d_i$，另一种是$d_i\ast i$

---

代码

#include<iostream>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;

int n,m,x,y,k,z;
long long tree[1000100],tree2[1000100],a[1000100];

void add(int x,long long y){
	for(int i=x;i<=n;i+=i&(-i))
	{
		tree[i]+=y;
		tree2[i]+=y*x;
	}
}

long long sum(int x){
	long long ans=0;
	for(int i=x;i>0;i-=i&(-i))
		ans+=tree[i]*(x+1)-tree2[i];
	return ans;
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
		add(i,a[i]);
		add(i+1,-a[i]);
	}
		
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&k);
		if(k==1)
		{
			scanf("%d%d%d",&x,&y,&z);
			add(x,z);
			add(y+1,-z);
		}
		if(k==2)
		{
			scanf("%d%d",&x,&y);
			printf("%lld\n",sum(y)-sum(x-1));
		}
	}
}