HDU 1166 树状数组&&线段树模板1

累计sum值。单点更新、区间查询。

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以内。

树状数组：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 50000;
int tree[maxn + 10];
char op[10];
inline int lowbit(int x) {
    return (x & -x);
}
void add(int x, int val) {
    while(x <= maxn) {
        tree[x] += val;
        x += lowbit(x);
    }
}
int get(int x) {
    int sum = 0;
    while(x) {
        sum += tree[x];
        x -= lowbit(x);
    }
    return sum;
}
int main() {
    int T;
    scanf("%d", &T);
    for(int cas = 1; cas <= T; ++ cas) {
        printf("Case %d:\n", cas);
        memset(tree, 0, sizeof(tree));
        int n, x, y, val;
        scanf("%d", &n);
        for(int i = 1; i <= n; ++ i)
            scanf("%d", &val), add(i, val);
        while(1) {
            scanf("%s", &op);
            if(op[0] == 'Q') scanf("%d%d", &x, &y), printf("%d\n", get(y) - get(x-1));
            else if(op[0] == 'A') scanf("%d%d", &x, &val), add(x, val);
            else if(op[0] == 'S') scanf("%d%d", &x, &val), add(x, -val);
            else if(op[0] == 'E') break;
        }
    }
}

线段树：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 50000;
int tree[maxn * 4];
int a[maxn + 5];
void create(int o, int L, int R) {
    if(L == R) {
        tree[o] = a[L];
        return;
    }
    int mid = L + R >> 1;
    create(o * 2, L, mid);
    create(o * 2 + 1, mid+1, R);
    tree[o] = tree[o * 2] + tree[o * 2 + 1];
}
int query(int o, int L, int R, int ql, int qr) {
    //越界返回
    if(ql > R || qr < L) return 0;
    //包含返回
    if(ql <= L && R <= qr) return tree[o];
    //累加左右孩子并返回
    int mid = L + R >> 1;
    int ans = 0;
    ans += query(o * 2, L, mid, ql, qr);
    ans += query(o * 2 + 1, mid + 1, R, ql, qr);
    return ans;
}
void update(int o, int L, int R, int pos, int val) {
    //越界返回
    if(pos < L || pos > R) return;
    //点更新
    if(L == R && L == pos) {
        tree[o] += val;
        return;
    }
    //向下更新
    int mid = L + R >> 1;
    update(o * 2, L, mid, pos, val);
    update(o * 2 + 1, mid + 1, R, pos, val);
    tree[o] = tree[o * 2] + tree[o * 2 + 1];
}
int main() {
    int T;
    char op[10];
    scanf("%d", &T);
    for(int cas = 1; cas <= T; ++ cas) {
        printf("Case %d:\n", cas);
        memset(tree, 0, sizeof(tree));
        int n, x, y, val;
        scanf("%d", &n);
        for(int i = 1; i <= n; ++ i)
            scanf("%d", a + i);
        create(1, 1, n);
        while(1) {
            scanf("%s", &op);
            if(op[0] == 'Q') scanf("%d%d", &x, &y), printf("%d\n", query(1, 1, n, x, y));
            else if(op[0] == 'A') scanf("%d%d", &x, &val), update(1, 1, n, x, val);
            else if(op[0] == 'S') scanf("%d%d", &x, &val), update(1, 1, n, x, -val);
            else if(op[0] == 'E') break;
        }
    }
}