#include <stdio.h>
#include <math.h>
#include <string.h>
typedef struct STU{
int grade;
int left;
int right;
}STU;
STU st[2097152];
int num[2000001];
int max(int a,int b)
{
return a>b?a:b;
}
void BuildTree(int i,int left,int right)
{
st[i].left=left;
st[i].right=right;
st[i].grade=0;//初始化为0
if (left==right)//如果是一个点,直接打上记号
{
num[left]=i;
return;
}
BuildTree(i<<1,left,(int)floor((left+right)/2.0));
BuildTree((i<<1)+1,(int)floor((left+right)/2.0)+1,right);//注意括号和优先级
}
void UpdataTree(int r,int b)
{
if (r==1)
{
st[r].grade+=b;
return;
}
st[r].grade=st[r].grade+b;
UpdataTree(r/2,b);
}
int sum;
void Query(int i,int l,int r)
{
if ((st[i].left==l)&&(st[i].right==r))
{
sum=sum+st[i].grade;
return;
}
i=i<<1;
if (l<=st[i].right)//如果左边有涉及
{
if (r<=st[i].right)//如果全在左部区间
Query(i,l,r);
else//不全在左区间
Query(i,l,st[i].right);//继续查询在左区间的那部分
}
i++;
if (r>=st[i].left)//如果右边有涉及
{
if (l>=st[i].left)
Query(i,l,r);//全在右区间
else
Query(i,st[i].left,r);
}
}
int main()
{
int T,p;
int i,j,n,m,c,d;
char s[10];
scanf("%d",&T);
p=T;
while(T--)
{
printf("Case %d:\n",p-T);
scanf("%d",&n);
BuildTree(1,1,n);
for (i=1;i<=n;i++)
{
scanf("%d",&m);
UpdataTree(num[i],m);
}
while(scanf("%s",s)&&(strcmp(s,"End")!=0))
{
scanf("%d%d",&c,&d);
if (strcmp(s,"Query")==0)
{
sum=0;
Query(1,c,d);
printf("%d\n",sum);
}
else
if (strcmp(s,"Add")==0)
{
UpdataTree(num[c],d);
}
else
UpdataTree(num[c],-1*d);
}
}
return 0;
}
线段树(hdu1166)
最新推荐文章于 2019-04-13 15:43:06 发布