模板【洛谷P3368】 【模板】树状数组 2

P3368 【模板】树状数组 2

如题，已知一个数列，你需要进行下面两种操作：

1.将某区间每一个数数加上x

2.求出某一个数的值

树状数组区间加，单点查询。

code：

#include <iostream>
#include <cstdio>

using namespace std;

const int wx=500017;

inline int read(){
	int sum=0,f=1; char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
	while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0'; ch=getchar();}
	return sum*f;
}

int sum[wx],n,m;

void add(int pos,int k){
	for(int i=pos;i<=n;i+=(i&-i))
		sum[i]+=k;
}

int query(int pos){
	int re=0;
	for(int i=pos;i>=1;i-=(i&-i))
		re+=sum[i];
	return re;
}

void add_wx(int l,int r,int k){
	add(l,k); add(r+1,-k);
}

int main(){
	n=read(); m=read();
	for(int i=1;i<=n;i++){
		int x; x=read();
		add(i,x); add(i+1,-x);
	}
	for(int i=1;i<=m;i++){
		int opt; opt=read();
		if(opt==1){
			int x,y,k;
			x=read(); y=read(); k=read();
			add_wx(x,y,k);
		}
		else{
			int x; x=read();
			printf("%d\n",query(x));
		}
	}
}