线段树做法
#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
const int N=50005;
int ans;
struct node
{
int l,r,sum;
int mid()
{ return (l+r)>>1 ; }
}tree[N*4];
void build(int l,int r,int rt)
{
tree[rt].l=l;
tree[rt].r=r;
if(l==r)
{ scanf("%d",&tree[rt].sum) ;
return ;
}
int m=tree[rt].mid();
build(l,m,rt<<1 );
build(m+1,r,rt<<1|1);
tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum;
}
void query(int l,int r,int rt,int L,int R)
{ if(L<=l&&R>=r)
{ ans+=tree[rt].sum;
return ;
}
int m=tree[rt].mid();
if(R<=m)
query(l,m,rt<<1,L,R);
else if(L>m)
query(m+1,r,rt<<1|1,L,R);
else
{ query(l,m,rt<<1,L,R);
query(m+1,r,rt<<1|1,L,R);
}
}
void updata(int l,int r,int rt,int pos,int add)
{ if(l==r)
{ tree[rt].sum+=add;
return ;
}
int m=tree[rt].mid();
if(pos<=m)
updata(l,m,rt<<1,pos,add);
else
updata(m+1,r,rt<<1|1,pos,add);
tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum;
}
int main()
{
int t,n;
int a,b;
char str[10];
scanf("%d",&t);
for(int i=1;i<=t;i++)
{ printf("Case %d:\n",i);
scanf("%d",&n);
build(1,n,1);
while( scanf("%s",&str))
{
if(str[0]=='E')
break;
scanf("%d%d",&a,&b);
if(str[0]=='Q')
{ ans=0;
query(1,n,1,a,b);
printf("%d\n",ans);
}
else if (str[0]=='A')
updata(1,n,1,a,b);
else
updata(1,n,1,a,-b);
}
}
return 0;
}
ZKW线段树
#include <iostream>
#include <stdio.h>
#include <memory.h>
using namespace std;
const int N=50005;
int T[4*N];
int M;
int PushUp(int n)
{
T[n]=T[n<<1]+T[n<<1|1];
}
void build(int n)
{
for(M=1;M<=n+1;M<<=1);
for(int i=M+1;i<=M+n;i++)
scanf("%d",&T[i]);
for(int i=M-1;i>0;i--)
PushUp(i);
}
void add(int n,int v)
{
for(T[n+=M]+=v,n>>=1;n;n>>=1)
PushUp(n);
}
int Query(int s,int t)
{
int ans=0;
for(s=s+M-1,t=t+M+1;s^t^1;s>>=1,t>>=1)
{
if(~s&1) ans+=T[s^1];
if(t&1) ans+=T[t^1];
}
return ans;
}
int main()
{
char str[10];
int n,t,a,b,k=1;
scanf("%d",&t);
for(int i=1;i<=t;i++)
{
printf("Case %d:\n",i);
memset(T,0,sizeof(T));
scanf("%d",&n);
build(n);
while(1)
{
scanf("%s",str);
if(str[0]=='E')
break;
scanf("%d%d",&a,&b);
if(str[0]=='A')
add(a,b);
else if(str[0]=='S')
add(a,-b);
else
printf("%d\n",Query(a,b));
}
}
}