洛谷 P3368 树状数组2

题目大意

给出一个数组，有两种操作，一种是把一个区间内的数全部加上k，一种是输出第x个数的值。

题目分析

直接暴力操作，肯定会超时，我们需要树状数组或者简单线段树来优化。

Code

#include<iostream>
#include<cstdio>
#define sco 500010
using namespace std;
int n,m,f,x,y,z,ans,a[sco];
struct tree{
	int l,r,data,lz;
}tr[sco*4];
void updata(int i){tr[i].data=tr[i*2].data+tr[i*2+1].data;}
void down(int i){
	if(tr[i].lz!=0){
		tr[i*2].lz+=tr[i].lz;
		tr[i*2+1].lz+=tr[i].lz;
		tr[i*2].data+=tr[i].lz*(tr[i*2].r-tr[i*2].l+1);
		tr[i*2+1].data+=tr[i].lz*(tr[i*2+1].r-tr[i*2+1].l+1);
		tr[i].lz=0;
	}
}
void build(int i,int l,int r){
	tr[i].l=l,tr[i].r=r;
	if(l==r){
		tr[i].data=a[l];return;
	}
	int mid((l+r)>>1);
	build(i*2,l,mid);
	build(i*2+1,mid+1,r);
	updata(i);
}
void modi(int i,int l,int r,int k){
	if(l<=tr[i].l && tr[i].r<=r){
		tr[i].data+=k;tr[i].lz+=k;return;
	}
	if(tr[i*2].r>=l)modi(i*2,l,r,k);
	if(tr[i*2+1].l<=r)modi(i*2+1,l,r,k);
	updata(i);
}
int search(int i,int l,int r){
	if(l<=tr[i].l && tr[i].r<=r){
		return tr[i].data;
	}
	if(tr[i].l>r || tr[i].r<l)return 0;
	down(i);
	int s(0);
	if(tr[i*2].r>=l)s+=search(i*2,l,r);
	if(tr[i*2+1].l<=r)s+=search(i*2+1,l,r);
	return s;
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i){
		scanf("%d",&a[i]);
	}
	build(1,1,n);
	for(int i=1;i<=m;++i){
		scanf("%d",&f);
		if(f==1){
			scanf("%d%d%d",&x,&y,&z);
			modi(1,x,y,z);continue;
		}
		scanf("%d",&x);
		ans=0;
		printf("%d\n",search(1,x,x));
	}
	return 0;
}