【BZOJ 1176 2683】Mokia

Description

维护一个W*W的矩阵，初始值均为S.每次操作可以增加某格子的权值,或询问某子矩阵的总权值.修改操作数M<=160000,询问数Q<=10000,W<=2000000.（在2683中，没有S，初始值为0）。

Input

第一行两个整数,S,W;其中S为矩阵初始值;W为矩阵大小

接下来每行为一下三种输入之一(不包含引号):

"1 x y a"

"2 x1 y1 x2 y2"

"3"

输入1:你需要把(x,y)(第x行第y列)的格子权值增加a

输入2:你需要求出以左上角为(x1,y1),右下角为(x2,y2)的矩阵内所有格子的权值和,并输出

输入3:表示输入结束

Output

对于每个输入2,输出一行,即输入2的答案

Sample Input

0 4
1 2 3 3
2 1 1 3 3
1 2 2 2
2 2 2 3 4
3

Sample Output

3
5

HINT 

保证答案不会超过int范围

 

分析：

　　对操作进行分治，左边的修改必定会影响右边的查询，这一部分只要对x排序，做树状数组就好了。

 

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>

struct A
{
    int x, y, pos, ctrl, ans;
} t[200010], q[200010];

inline bool cmp(const A &a, const A &b)
{
    return a.pos < b.pos;
}

int s, w, k, x, y, a, b, cnt;
int key[2000010], late[2000010], now;

void modify(int pos, int data)
{
    for ( ; pos <= w + 1; pos += pos & -pos)
    {
        if (late[pos] != now) key[pos] = 0;
        late[pos] = now, key[pos] += data;
    }
}

int query(int pos)
{
    int ret = 0;
    for ( ; pos; pos -= pos & -pos)
        if (late[pos] == now) ret += key[pos];
    return ret;
}

void cdq(int left, int right)
{
    if (left == right) return;
    int mid = left + right >> 1;
    cdq(left, mid); cdq(mid + 1, right);
    now++;
    for (int i = mid + 1, j = left; i <= right; i++)
    {
        for ( ; j <= mid && q[j].x <= q[i].x; j++)
            if (q[j].ctrl == 0) modify(q[j].y, q[j].ans);
        if (q[i].ctrl > 0) q[i].ans += query(q[i].y);
    }
    for (int i = left, j = left, k = mid + 1; i <= right; i++)
    {
        if ((q[j].x <= q[k].x && j <= mid) || k > right)
            t[i] = q[j++];
        else t[i] = q[k++];
    }
    memcpy(q + left, t + left, sizeof(q[0]) * (right - left + 1));
}

int main()
{
    scanf("%d%d", &s, &w);
    while (k != 3)
    {
        scanf("%d", &k);
        if (k == 1)
        {
            scanf("%d%d%d", &x, &y, &a);
            cnt++, q[cnt] = (A) {x + 1, y + 1, cnt, 0, a};
        }
        if (k == 2)
        {
            scanf("%d%d%d%d", &x, &y, &a, &b);
            a++, b++;
            cnt++, q[cnt] = (A) {x, y, cnt, 1, x * y * s};
            cnt++, q[cnt] = (A) {a, b, cnt, 2, a * b * s};
            cnt++, q[cnt] = (A) {a, y, cnt, 2, a * y * s};
            cnt++, q[cnt] = (A) {x, b, cnt, 2, x * b * s};
        }
    }
    cdq(1, cnt);
    std::sort(q + 1, q + cnt + 1, cmp);
    for (int i = 1; i <= cnt; i++)
    {
        if (q[i].ctrl == 1)
        {
            printf("%d\n", q[i].ans + q[i + 1].ans - q[i + 2].ans - q[i + 3].ans);
        }
    }
}