题目链接
简单的线段树入门题(终于入门,线段树是个好东西)
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxn = 5e4+10;
struct ST
{
int l,r;
int sum;
} t[maxn*4];
int a[maxn*4];
int ans;
void build(int p,int l,int r)
{
t[p].l = l;
t[p].r = r;
if(l == r)
{
scanf("%d",&t[p].sum);
return ;
}
int mid = (l+r)/2;
build(p*2,l,mid);//zuo
build(p*2+1,mid+1,r);//you
t[p].sum = t[p*2].sum+t[p*2+1].sum;
}
void updata(int p,int x,int v,int flg)
{
t[p].sum = t[p].sum+flg*v;
if(t[p].l == t[p].r)
return ;
int mid = (t[p].l+t[p].r)/2;
if(x <= mid)
updata(p*2,x,v,flg);
else
updata(p*2+1,x,v,flg);
}
void ask(int p,int l,int r)
{
if(l <= t[p].l && r >= t[p].r)
ans += t[p].sum;
else
{
int mid = (t[p].l+t[p].r)/2;
if(r <= mid)
ask(p*2,l,r);
else if(l > mid)
ask(p*2+1,l,r);
else
{
ask(p*2,l,r);
ask(p*2+1,l,r);
}
}
}
char str[10];
int main()
{
int t;
scanf("%d",&t);
int num = 1;
while(t--)
{
printf("Case %d:\n",num++);
int n,x,v;
scanf("%d",&n);
// for(int i = 0; i < n; ++i)
//scanf("%d",&a[i]);
build(1,1,n);
scanf("%s",str);
while(1)
{
if(strcmp(str,"End") == 0)
break;
else if(strcmp(str,"Add") == 0)
{
scanf("%d%d",&x,&v);
updata(1,x,v,1);
}
else if(strcmp(str,"Sub") == 0)
{
scanf("%d%d",&x,&v);
updata(1,x,v,-1);
}
else
{
ans = 0;
scanf("%d%d",&x,&v);
ask(1,x,v);
printf("%d\n",ans);
}
scanf("%s",str);
}
}
return 0;
}