敌兵布阵 - 线段树讲解（改变点的值）

C国的死对头A国这段时间正在进行军事演习，所以C国间谍头子Derek和他手下Tidy又开始忙乎了。A国在海岸线沿直线布置了N个工兵营地,Derek和Tidy的任务就是要监视这些工兵营地的活动情况。由于采取了某种先进的监测手段，所以每个工兵营地的人数C国都掌握的一清二楚,每个工兵营地的人数都有可能发生变动，可能增加或减少若干人手,但这些都逃不过C国的监视。 
中央情报局要研究敌人究竟演习什么战术,所以Tidy要随时向Derek汇报某一段连续的工兵营地一共有多少人,例如Derek问:“Tidy,马上汇报第3个营地到第10个营地共有多少人!”Tidy就要马上开始计算这一段的总人数并汇报。但敌兵营地的人数经常变动，而Derek每次询问的段都不一样，所以Tidy不得不每次都一个一个营地的去数，很快就精疲力尽了，Derek对Tidy的计算速度越来越不满:"你个死肥仔，算得这么慢，我炒你鱿鱼!”Tidy想：“你自己来算算看，这可真是一项累人的工作!我恨不得你炒我鱿鱼呢!”无奈之下，Tidy只好打电话向计算机专家Windbreaker求救,Windbreaker说：“死肥仔，叫你平时做多点acm题和看多点算法书，现在尝到苦果了吧!”Tidy说："我知错了。。。"但Windbreaker已经挂掉电话了。Tidy很苦恼，这么算他真的会崩溃的，聪明的读者，你能写个程序帮他完成这项工作吗？不过如果你的程序效率不够高的话，Tidy还是会受到Derek的责骂的. 

Input第一行一个整数T，表示有T组数据。 
每组数据第一行一个正整数N（N<=50000）,表示敌人有N个工兵营地，接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人（1<=ai<=50）。 
接下来每行有一条命令，命令有4种形式： 
(1) Add i j,i和j为正整数,表示第i个营地增加j个人（j不超过30） 
(2)Sub i j ,i和j为正整数,表示第i个营地减少j个人（j不超过30）; 
(3)Query i j ,i和j为正整数,i<=j，表示询问第i到第j个营地的总人数; 
(4)End 表示结束，这条命令在每组数据最后出现; 
每组数据最多有40000条命令 
Output对第i组数据,首先输出“Case i:”和回车, 
对于每个Query询问，输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。 
Sample Input

1
10
1 2 3 4 5 6 7 8 9 10
Query 1 3
Add 3 6
Query 2 7
Sub 10 2
Add 6 3
Query 3 10
End 

Sample Output

Case 1:
6
33
59

分析：

从这题开始学习线段树……看这道题的时候迷之眼熟啊，然后发现我写过这题的题解……大概在半年前，貌似学长讲过……然后当时没听懂，写题解的时候就写了一句话：这是学长的代码，我只是学习一下……

（图盗）

线段树利用了树形结构的性质，使根节点对应区间1到n的加和，其左孩子对应从1到m的加和，右孩子对应从m+1到n的加和，m是1到n的中点，m=(1+n)/2。

生成线段树：

/*
l与r是题中给的数组的下标
k是我们建立树状数组的下标
*/
void build(int l,int r,int k){
    if(l==r){
        sum[k]=a[l];
        return ;
    }
    int m=(l+r)>>1;
    build(l,m,k<<1);//当前结点的左孩子序号是当前结点序号*2
    build(m+1,r,k<<1|1);//当前结点的右孩子序号是当前结点序号*2+1

    sum[k]=sum[k<<1]+sum[k<<1|1];//左右孩子相加
}

查找L到R的区间和：

1）如果现在所在的区间全部包含在[L,R]中，则返回其区间

2）否则，从中间分割，向下递归，如图，求[3,12]区间的和

/*
l与r是题中给的数组的下标
L与R是我们要求的区间的下标
k是我们建立树状数组的下标
*/
int query(int L,int R,int l,int r,int k){
    if(L<=l&&R>=r)return sum[k];
    int ans=0;
    int m=(l+r)>>1;
    if(L<=m)ans+=query(L,R,l,m,k<<1);//当前区间的左端点小于要求区间的左端点，则向左走一步
    if(R>m)ans+=query(L,R,m+1,r,k<<1|1);
    return ans;
}

注意：是L<=m,R>m，枚举一下便可知为什么这样写。

进行点修改：

这与上一步的代码非常相似了：

/*
P我们要修改的点在数组a中的下标
C增量
l,r记录区间
k是我们建立树状数组的下标
*/
void update(int P,int C,int l,int r,int k){
     if(l==r){sum[k]+=C;return ;}
     int m=(l+r)>>1;
     if(P<=m)update(P,C,l,m,k<<1);
     if(P>m)update(P,C,m+1,r,k<<1|1);
     sum[k]=sum[k<<1]+sum[k<<1|1];
}

注意：数组sum要开4倍大啊，不然会tle的

代码如下：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
int sum[200003],a[50003];
using namespace std;

void build(int l,int r,int k){
    if(l==r){
        sum[k]=a[l];
        return ;
    }
    int m=(l+r)>>1;
    build(l,m,k<<1);
    build(m+1,r,k<<1|1);
    sum[k]=sum[k<<1]+sum[k<<1|1];
}

int query(int L,int R,int l,int r,int k){
    if(L<=l&&R>=r)return sum[k];
    int m=(l+r)>>1;//从中间劈开
    int ans=0;
    if(L<=m)ans+=query(L,R,l,m,k<<1);//向左走一个结点
    if(R>m)ans+=query(L,R,m+1,r,k<<1|1);//向右走一个结点
    return ans;
}

void update(int P,int C,int l,int r,int k){
    if(l==r){
        sum[k]+=C;
        return ;
    }
    int m=(l+r)>>1;
    if(P<=m)update(P,C,l,m,k<<1);
    else update(P,C,m+1,r,k<<1|1);
    sum[k]=sum[k<<1]+sum[k<<1|1];
}

int main(){
    int T,N,n,k,m;
    char s[5];
    scanf("%d",&T);
    for(int q=1;q<=T;q++){
        printf("Case %d:\n",q);
        scanf("%d",&N);
        memset(a,0,sizeof(a));
        memset(sum,0,sizeof(sum));

        for(int i=1;i<=N;i++){
            scanf("%d",&a[i]);
        }

        build(1,N,1);

        while(scanf("%s",s)){
            if(s[0]=='E')break;
            scanf("%d%d",&n,&m);
            if(s[0]=='Q'){//询问区间[n,m]
                printf("%d\n",query(n,m,1,N,1));
                continue;
            }
            if(s[0]=='S')m=-m;

            update(n,m,1,N,1);
        }
    }
}