线段树求区间和(单点更新)

题目1：敌兵布阵 

 

线段树的主要操作：（1）建立线段树（Build）          （2）更新区间值 （Update）           （3）查询区间（Query）

写法一：

#include 
#define maxn 55555

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

int sum[maxn<<2];

void PushUP(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void Build(int l,int r,int rt)
{
    if(l==r)
    {
        scanf("%d",&sum[rt]);
        return;
    }
    int m=(l+r)>>1;
    Build(lson);
    Build(rson);
    PushUP(rt);
}

void Update(int p,int add,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]+=add;
        return;
    }
    int m=(l+r)>>1;
    if(p<=m)
        Update(p,add,lson);
    else
        Update(p,add,rson);
    PushUP(rt);
}

int Query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r)
         return sum[rt];
    int m=(l+r)>>1;
    int ret=0;
    if(L<=m)    ret+=Query(L,R,lson);
    if(R>m)     ret+=Query(L,R,rson);
    return ret;
}

int main()
{
    int n,t,k=1;
    char s[15];
    scanf("%d",&t);
    while(t--)
    {
        printf("Case %d:\n",k++);
        scanf("%d",&n);
        Build(1,n,1);
        while(~scanf("%s",s))
        {
            if(s[0]=='E')
                break;
            int a,b;
            scanf("%d%d",&a,&b);
            if(s[0]=='Q')
                printf("%d\n",Query(a,b,1,n,1));
            else if(s[0]=='S')
                Update(a,-b,1,n,1);
            else
                Update(a,b,1,n,1);
        }
    }
    return 0;
}

 

写法二：

#include
#define N 50005

typedef struct
{
    int Lson;
    int Rson;
    int sum;
}Tree;

Tree seg_tree[N*3];
int a[N];

void Build(int num,int L,int R)
{
    seg_tree[num].Lson=L;
    seg_tree[num].Rson=R;

    if(L==R)
    {
        seg_tree[num].sum=a[L];
        return;
    }
    else
    {
        int mid=(L+R)>>1;
        Build(2*num,L,mid);
        Build(2*num+1,mid+1,R);
        seg_tree[num].sum=seg_tree[2*num].sum+seg_tree[2*num+1].sum;
    }
}

void Update(int num,int L,int R,int a,int b)
{
    seg_tree[num].sum+=b;
    if(L==R&&L==a)
        return;

    int mid=(L+R)>>1;
    if(a<=mid)
        Update(2*num,L,mid,a,b);
    else
        Update(2*num+1,mid+1,R,a,b);
}

int Query(int num,int L,int R,int a,int b)
{
    if(L==a&&R==b)
        return seg_tree[num].sum;
    int mid=(L+R)>>1;
    if(b<=mid)
        return Query(2*num,L,mid,a,b);
    if(a>=mid+1)
        return Query(2*num+1,mid+1,R,a,b);
    return Query(2*num,L,mid,a,mid)+Query(2*num+1,mid+1,R,mid+1,b);
}

int main()
{
    int t,n,k;
    char str[15];
    scanf("%d",&t);

    for(k=1;k<=t;k++)
    {
        printf("Case %d:\n",k);
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        Build(1,1,n);
        while(~scanf("%s",str))
        {
            int a,b;
            if(str[0]=='E')
                break;
            scanf("%d%d",&a,&b);
            if(str[0]=='A')
                Update(1,1,n,a,b);
            else if(str[0]=='S')
                Update(1,1,n,a,-b);
            else
                printf("%d\n",Query(1,1,n,a,b));
        }
    }
    return 0;
}