敌兵布阵

这道题,题意比较简单,原本打算一路上全部for过去的,然而是我想太多,不断超时,只能使用线段树,线段树的话,我可能讲不清,所以就直接贴代码吧。可以好好查资料的说,很多博客里面都会有详解。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
const int maxn = 55555;
int hua[maxn << 2];
void PushUp(int rt)
{
    hua[rt] = hua[rt << 1] + hua[rt << 1 | 1];
}

void build(int l, int r, int rt)
{
    if (l == r)
    {
        scanf("%d", &hua[rt]);
        return;
    }
    int m = (l + r) >> 1;
    build(l, m, rt << 1);
    build(m + 1, r, rt << 1 | 1);
    PushUp(rt);
}

void update(int p, int add, int l, int r, int rt)
{
    if (l == r)
    {
        hua[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);
    PushUp(rt);
}

int query(int ll, int rr, int l, int r, int rt)
{
    if (ll <= l && rr >= r) return hua[rt];
    int m = (l + r) >> 1;
    int ret = 0;
    if (ll <= m) ret += query(ll, rr, l, m, rt << 1);
    if (rr > m) ret += query(ll, rr, m + 1, r, rt << 1 | 1);
    return ret;
}

int main()
{
    int t, c;
    char d[10];
    scanf("%d", &t);
    for (c = 1; c <= t; c++)
    {
        printf("Case %d:\n", c);
        int n;
        scanf("%d", &n);
        build(1, n, 1);
        while (scanf("%s", d) != EOF)
        {
            if (d[0] == 'E') break;
            int x, y;
            scanf("%d%d", &x, &y);
            if (d[0] == 'Q')
            {
                int ans = query(x, y, 1, n, 1);
                printf("%d\n", ans);
            }
            if (d[0] == 'S') update(x, -y, 1, n, 1);
            if (d[0] == 'A') update(x, y, 1, n, 1);
        }
    }
    return 0;
}

  

posted @ 2016-07-22 18:53  阿银家的小贰  阅读(112)  评论(0编辑  收藏  举报