前往:我自己搭建的博客
 
题目
 
洛谷P3368树状数组2
 
题解
 
将普通树状数组操作和差分思想结合,就可以实现区间修改和单点查询。若对区间[l,r]进行修改,则在l处修改,并在r+1处反向修改。在查询时,询问x处的值,则求[1,x]的和,若x在[l,r]内,则会累加l处的修改,若x在r右侧,则r+1处的反向修改会抵消l处的修改。
 
代码
 
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=5e5+5;
int n,m;
int sum[maxn],a[maxn];	//a[]存初始数据 
inline int lowbit(int x) {return x&(-x);}
inline void add(int x,int y) {for( ;x<=n;x+=lowbit(x)) sum[x]+=y;}
inline int ask(int x) {int ans=0; for( ;x;x-=lowbit(x)) ans+=sum[x]; return ans;}
			
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<=m;i++)
	{
		int t; scanf("%d",&t);
		if(t==1)
		{
			int x,y,k; scanf("%d%d%d",&x,&y,&k);
			add(x,k); add(y+1,-k);
		}
		else
		{
			int x; scanf("%d",&x);
			printf("%d\n",a[x]+ask(x));	//注意加上原始数据 
		}
	} 
	
	return 0;
}