POJ 1195 - Moblie phones

—— by A Code Rabbit

Description

在一个矩阵上面进行一些操作。

将矩阵某一个位置上的数字增加，

或者是询问某个子矩阵内的数值之和。

对于每个询问，输出结果。

Type

Advanced Data Structures :: Segment Tree

Advanced Data Structures :: Binary Indexed Tree

Analysis

经典的二维树状数组。

只是将一维 BIT 变成二维 BIT 的过程而已。

然后求和的时候，利用类似集合操作的方法求和。

还是要注意 BIT 下标不能为 0 的问题。

Solution

// POJ 1195// Mobile phones// by A Code Rabbit#include <cstdio>#include <cstring>const int MAXN = 1026;const int MAXM = 1026;struct Bit2D {    int c[MAXN][MAXM], n, m;    void Init(int x, int y) { memset(c, 0, sizeof(c)); n = x; m = y; }    void Add(int x, int y, int z) {        int tmp = y;        while (x <= n) {            y = tmp;            while (y <= m) {                c[x][y] += z;                y += y & -y;            }            x += x & -x;        }    }    int Sum(int x, int y) {        int res = 0;        int tmp = y;        while (x > 0) {            y = tmp;            while (y > 0) {                res += c[x][y];                y -= y & -y;            }            x -= x & -x;        }        return res;    }};int instruction, s;int x, y, a;int l, b, r, t;Bit2D bit;int main() {    while (scanf("%d", &instruction)) {        if (instruction == 0) {            scanf("%d", &s);            bit.Init(s, s);        } else        if (instruction == 1) {            scanf("%d%d%d", &x, &y, &a);            bit.Add(x + 1, y + 1, a);        } else        if (instruction == 2) {            scanf("%d%d%d%d", &l, &b, &r, &t);            int ans = bit.Sum(r + 1, t + 1)                - bit.Sum(l, t + 1)                - bit.Sum(r + 1, b)                + bit.Sum(l, b);            printf("%d\n", ans);        } else        if (instruction == 3) {            break;        }    }    return 0;}