hud 1166

线段树做法

#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));
        }
    }
}