HDU 1166 敌兵布阵（线段树）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1166

题       意：给你一组数据，进行下列操作：

                接下来每行有一条命令，命令有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  表示结束，这条命令在每组数据最后出现;

思       路：简单的线段树，直接套用模板进行了。

代码如下：

#include <iostream>
using namespace std;
#include <string.h>
#include <stdio.h>
#include <queue>
#include <algorithm>
#define m1 300000
int vis[m1];
void build( int l, int r ,int rt )//rt编号
{
    if( l == r )
    {
        scanf ( "%d", &vis[rt] );
        return ;
    }
    int m = ( l + r ) >> 1;
    build( l, m, rt<<1 );
    build( m+1, r, rt<<1|1);
    vis[rt]=vis[rt<<1]+vis[rt<<1|1];
}
void update( int p, int add, int l, int r, int rt )
{
    if( l == r )
    {
        vis[rt] += add;
        return ;
    }
    int m = ( l + r ) >> 1;
    if( p <= m ) update(p,add,l,m,rt<<1);
    else update(p,add,m+1,r,rt<<1|1);
    vis[rt]=vis[rt<<1]+vis[rt<<1|1];
}
int query( int L, int R, int l, int r, int rt )
{
    if( L <= l && r <= R )
        return vis[rt];
    int m = ( l + r ) >> 1;
    int ret = 0;
    if( L <= m ) ret += query(L,R,l,m,rt<<1);
    if( R > m ) ret += query(L,R,m+1,r,rt<<1|1);
    return ret;
}
int main()
{
    int T;
    scanf ( "%d", &T );
    int ans = 1;
    while ( T-- )
    {
        int n;
        printf("Case %d:\n",ans++);
        scanf ( "%d", &n );
        build(1,n,1);
        char str[10];
        while( scanf ( "%s", str ) != EOF )
        {
            if( str[0] == 'E' ) break;
            int x, y;
                scanf ( "%d %d", &x, &y );
            if( str[0] == 'A' ) {
                update(x,y,1,n,1);
            }
            if( str[0] == 'S' ) {
                update(x,-y,1,n,1);
            }
            if( str[0] == 'Q' ){
                printf("%d\n",query(x,y,1,n,1));
            }
        }
    }
    return 0;
}