HDU-1166 敌兵布阵

来源：互联网发布：windows 挂载编辑：程序博客网时间：2024/06/03 16:19

#include <iostream>#include <cstdio>using namespace std;#define lson l, m, rt << 1#define rson m + 1, r, rt << 1 | 1const int maxn = 55555;int sum[maxn<<2];void PushUp(int rt){    sum[rt] = sum[rt<<1] + sum[rt<<1|1];}void build(int l, int r, int rt){    if(l == r)    {        scanf("%d",&sum[rt]);        return ;    }    int m = (l + r) >> 1;    build(lson);    build(rson);    PushUp(rt);}void update(int p, int add, int l, int r, int rt){    if(l == r)    {        sum[rt] += add;        return ;    }    int m = (l + r) >> 1;    if(p <= m) update(p, add, lson);    else update(p, add, rson);    PushUp(rt);}int query(int L, int R, int l, int r, int rt){    if(L <= l && r <= R)    {        return sum[rt];    }     int m = (l + r) >> 1;     int ret = 0;     if(L <= m) ret += query(L, R, lson);     if(R > m) ret += query(L, R, rson);     return ret;}int main(){    int T, n,cas = 0;    cin >> T;    while(T--)    {        cout << "Case " << ++cas << ":" << endl;        cin >> n;        build(1,n,1);        char op[10];        while(cin >> op)        {            if(op[0] == 'E')                break;            int a, b;            cin >> a >> b;            if(op[0] == 'Q')                cout << query(a, b, 1, n, 1) << endl;            else if(op[0] == 'S')                update(a, -b, 1, n, 1);            else                update(a, b, 1, n, 1);        }    }    return 0;}

题意：

第一行一个整数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条命令

输出query。

题解：建线段树吧..掌握模板 掌握模板..有点难有点难..开始学开始学。 函数PushUp更新父节点，函数build建树，函数pudate单点更新，函数query区间求和。

还有用cin会超时。

0 0