HDU1166 敌兵布阵

题目传送门

此题目是树状数组的模板题,没有任何技术含量,记忆吧~

#include <cstdio>
#include <cstring>
using namespace std;
const int N = 50010;
int n;

// 树状数组模板
#define lowbit(x) (x & -x)
typedef long long LL;
int c[N];
void add(int x, int v) {
    while (x < N) c[x] += v, x += lowbit(x);
}
LL sum(int x) {
    LL res = 0;
    while (x) res += c[x], x -= lowbit(x);
    return res;
}

int main() {
    int T;
    scanf("%d", &T);
    char op[110]; // 读入指令

    for (int k = 1; k <= T; k++) {
        scanf("%d", &n);
        // 多组测试数据,需要清空树状数组
        memset(c, 0, sizeof c);
        for (int i = 1; i <= n; i++) {
            int x;
            scanf("%d", &x);
            add(i, x); // i这个营地有x个士兵,映射概念:位置+数量
        }
        // 输出
        printf("Case %d:\n", k);
        while (scanf("%s", op) && op[0] != 'E') {
            int x, y;
            scanf("%d %d", &x, &y);
            if (op[0] == 'Q')
                printf("%lld\n", sum(y) - sum(x - 1)); // 计算区间内士兵个数,前缀和
            else if (op[0] == 'A')
                add(x, y); // 在x号营地增加y名士兵
            else
                add(x, -y); // 在x号营地减少y名士兵
        }
    }
    return 0;
}
posted @ 2022-05-04 17:19  糖豆爸爸  阅读(35)  评论(0)    收藏  举报
Live2D