以前用線段樹寫的,編碼複雜,效率也低。今天剛學習了樹狀數組,發現這道題可以用樹狀數組來寫,而且時間也節省了一半,編碼複雜度和空間複雜度就更不用說了。
#include<stdio.h>
#include<string.h>
#define lowbit(x) x&(-x)
int a[50010],n;
void add(int x,int d)
{
while(x<=n)
{
a[x]+=d;
x+=lowbit(x);
}
}
int q(int x)
{
int sum=0;
for(int i=x;i;i-=lowbit(i))
sum+=a[i];
return sum;
}
int main()
{
int t,i,tmp,l,r,cas=0;
char s[100];
scanf("%d",&t);
while(t--)
{
printf("Case %d:\n",++cas);
memset(a,0,sizeof(a));
scanf("%d",&n);
for(i=1;i<=n;i++)
{
scanf("%d",&tmp);
add(i,tmp);
}
getchar();
while(scanf("%s",s)!=EOF&&s[0]!='E')
{
scanf("%d%d",&l,&r);
if(s[0]=='A')
add(l,r);
else if(s[0]=='S')
add(l,-r);
else
printf("%d\n",q(r)-q(l-1));
}
}
return 0;
}