LightOJ1112——Curious Robin Hood

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>

using namespace std;

const int maxN=100010;
int tree[4*maxN];
int money[maxN];

void build(int no,int l,int r)
{
if(l==r)
{
tree[no]=money[l];
return;
}

int mid=(l+r)>>1;
int no2=no<<1;
build(no2,l,mid);
build(no2+1,mid+1,r);
tree[no]=tree[no2]+tree[no2+1];
}

void update1(int no,int l,int r,int x)
{
if(l==r)
{
printf("%d\n",tree[no]);
tree[no]=0;
return;
}

int mid=(l+r)>>1;
int no2=no<<1;
if(x<=mid) update1(no2,l,mid,x);
else update1(no2+1,mid+1,r,x);
tree[no]=tree[no2]+tree[no2+1];
}

void update2(int no,int l,int r,int x,int change)
{
if(l==r)
{
tree[no]+=change;
return;
}

int mid=(l+r)>>1;
int no2=no<<1;
if(x<=mid) update2(no2,l,mid,x,change);
else update2(no2+1,mid+1,r,x,change);
tree[no]=tree[no2]+tree[no2+1];
}

int query(int no,int l,int r,int a,int b)
{
if(a==l && b==r)
{
return tree[no];
}

int mid=(l+r)>>1;
int no2=no<<1;
if(b<=mid) return query(no2,l,mid,a,b);
if(a>mid) return query(no2+1,mid+1,r,a,b);
return query(no2,l,mid,a,mid)+query(no2+1,mid+1,r,mid+1,b);
}

int main()
{
int t;

//freopen("stdin.txt","r",stdin);
scanf("%d",&t);
for(int id=1; id<=t; id++)
{
int n,m;
int cas;
int x,y;

printf("Case %d:\n",id);
scanf("%d %d",&n,&m);
for(int i=1; i<=n; i++)
scanf("%d",&money[i]);
build(1,1,n);
for(int i=1; i<=m; i++)
{
scanf("%d",&cas);
switch(cas)
{
case 1:
scanf("%d",&x);
x+=1;
update1(1,1,n,x);
break;
case 2:
scanf("%d %d",&x,&y);
x+=1;
update2(1,1,n,x,y);
break;
case 3:
scanf("%d %d",&x,&y);
x+=1,y+=1;
printf("%d\n",query(1,1,n,x,y));
break;
}
}
}
return 0;
}