第一道线段树题,还是不理解,这是看了学长的代码敲出来的!
<0+0>
#include<stdio.h>
#define N 50005
struct node
{
int x,y,sum;
}a[3*N];
int b[N];
void CreatTree(int t,int x,int y)
{
a[t].x=x;
a[t].y=y;
a[t].sum=0;
if(x==y)
return;
int temp=2*t;
int mid=(x+y)/2;
CreatTree(temp,x,mid);
CreatTree(temp+1,mid+1,y);
return;
}
void InsertTree(int t,int x,int k)
{
if(a[t].x==a[t].y)
{
a[t].sum+=k;
return;
}
int temp=2*t;
int mid=(a[t].x+a[t].y)/2;
if(x<=mid)
InsertTree(temp,x,k);
else
InsertTree(temp+1,x,k);
a[t].sum=a[temp].sum+a[temp+1].sum;
return;
}
int FindTree(int t,int x,int y)
{
if(a[t].x==x&&a[t].y==y)
return a[t].sum;
int temp=t*2;
int mid=(a[t].x+a[t].y)/2;
if(y<=mid)
return FindTree(temp,x,y);
else if(x>mid)
return FindTree(temp+1,x,y);
else
return FindTree(temp,x,mid)+FindTree(temp+1,mid+1,y);
}
int main()
{
int T;
int cnt;
cnt=1;
scanf("%d",&T);
while(T--)
{
int n,i;
scanf("%d",&n);
CreatTree(1,1,n);
for(i=1;i<=n;i++)
{
scanf("%d",&b[i]);
InsertTree(1,i,b[i]);
}
char str[50];
int x,y;
printf("Case %d:\n",cnt);
while(scanf("%s",str),str[0]!='E')
{
scanf("%d%d",&x,&y);
if(str[0]=='Q')
printf("%d\n",FindTree(1,x,y));
else if(str[0]=='A')
InsertTree(1,x,y);
else
InsertTree(1,x,-y);
}
cnt++;
}
return 0;
}
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1166