hdu 1166 敌兵布阵（线段树模板题）

题目

---

第i个正整数ai代表第i个工兵营地里开始时有ai个人

(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条命令

---

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1e6+5;///
const int M = 10;
char str[M];
int ans[N];
int t, n, x, l, r, sum, cnt;

struct tree{
    int l, r;///左右区间
    int n;///值
}a[N];

void init()///建树
{
    int i, k = 1;
    while(k < n) k <<= 1;///
    for(i=k; i<2*k; i++){///最底层
        a[i].l = a[i].r = i-k+1;
        a[i].n = 0;
    }
    for(i=k-1; i>0; i--){///往上更新
        a[i].l = a[2*i].l;
        a[i].r = a[2*i+1].r;
        a[i].n = 0;
    }
//    printf("===%d\n", n);
//    for(i=0; i<k; i++)
//        printf("%d_%d_%d\n", a[i].l, a[i].r, a[i].n);
}

void inSert(int i, int x, int num)///x位置增加num人
{
    if(x>=a[i].l && x<=a[i].r)
        a[i].n += num;///
    if(a[i].l == a[i].r)
        return;
    int mid = (a[i].l + a[i].r) / 2;
    if(x > mid)///
        inSert(2*i+1, x, num);
    else
        inSert(2*i, x, num);
}

void Find(int x, int y, int i)///查询x~y区间的值之和
{
    if(x==a[i].l && y==a[i].r){
        sum += a[i].n;
        return;
    }
    if(a[i].l == a[i].r)
        return;
    int mid = (a[i].l+a[i].r)/2;
    if(x > mid)
        Find(x, y, 2*i+1);
    else if(y <= mid)
        Find(x, y, 2*i);
    else{
        Find(x, mid, 2*i);
        Find(mid+1, y, 2*i+1);
    }
}

int main()
{
    scanf("%d", &t);
    for(int m=0; m<t; m++){
        scanf("%d", &n);
        init();
        for(int i=1; i<=n; i++){
            scanf("%d", &x);
            inSert(1, i, x);
        }
        cnt = x = 0;
        while(scanf("%s", str)){
            if(str[0] == 'E')
                break;
            else if(str[0] == 'A'){///增加
                scanf("%d%d", &l, &x);///
                inSert(1, l, x);
            }
            else if(str[0] == 'S'){///减少
                scanf("%d%d", &l, &x);
                inSert(1, l, -x);
            }
            else if(str[0] == 'Q'){///询问
                scanf("%d%d", &l, &r);
                sum = 0;
                Find(l, r, 1);
                ans[cnt++] = sum;//printf("%d\n", sum);///保存结果
            }
        }
        printf("Case %d:\n", m+1);
        for(int i=0; i<cnt; i++){
            printf("%d\n", ans[i]);
        }
    }
    return 0;
}