【K-D树 求矩阵的和】BZOJ-4066 简单题

Problem Description

输入一个N，代表有一个N*N的棋盘。
有三种操作
1 x y A : 将格子x, y里的数字加上A
2 xl yl xr yr : 输出xl yl xr yr这个矩阵内的数字和（xl <= xr, yl <= yr）
3 : 终止条件
接下来每行一个操作。每条命令除第一个数字之外，
均要异或上一次输出的答案last_ans，初始时last_ans=0。
对于每个2操作，输出一个对应的答案。

思路：

学习ldq的思路
这里插入的点的数量有点多，可能导致树的形态太差，所以插入点树每过 5000 就暴力重构一下。
注意求矩阵和函数，和线段树区间求和 还是有点相似的。
其他函数基本不变

#include<bits/stdc++.h>using namespace std;const int MAX = 200055;const int inf = 0x3f3f3f3f;const int DIM = 2;struct node{    int l, r;    int sum, v;//sum记录这个子树的和，v代表该点的值    int d[DIM], maxn[DIM], minn[DIM];    inline void maintain()//初始化    {        for(int i = 0; i < DIM; i++)            maxn[i] = minn[i] = d[i];        l = r = 0;        sum = v;    }} tree[MAX*2];int D;bool operator < (const node &a, const node &b)//重载从小到大{    return a.d[D] < b.d[D];}inline void Merge(int o)//归并，向上更新{    int son[2] = {tree[o].l, tree[o].r};    for(int i = 0; i < 2; i++)    {        if(!son[i]) continue;        for(int j = 0; j < DIM; j++)        {            tree[o].maxn[j] = max(tree[o].maxn[j], tree[son[i]].maxn[j]);            tree[o].minn[j] = min(tree[o].minn[j], tree[son[i]].minn[j]);        }    }    tree[o].sum = tree[o].v + tree[son[0]].sum + tree[son[1]].sum;}int build(int l, int r, int now)//建树{    int mid = (l+r) >> 1;    D = now;    nth_element(tree+l, tree+mid, tree+r+1);    tree[mid].maintain();    if(l < mid) tree[mid].l = build(l, mid-1, (now+1)%DIM);    if(r > mid) tree[mid].r = build(mid+1, r, (now+1)%DIM);    Merge(mid);    return mid;}void Insert(int &o, int k, int now)//插入{    if(o == 0)    {        o = k;        return ;    }    if(tree[k].d[now] < tree[o].d[now]) Insert(tree[o].l, k, (now+1)%DIM);    else Insert(tree[o].r, k, (now+1)%DIM);    Merge(o);}int xl, xr, yl, yr;int ans;void query(int o)//求和{    if(xr < tree[o].minn[0] || xl > tree[o].maxn[0] || yl > tree[o].maxn[1] || yr < tree[o].minn[1])//这个子树，不在所求范围内        return;    if(xl <= tree[o].minn[0] && xr >= tree[o].maxn[0] && yl <= tree[o].minn[1] && yr >= tree[o].maxn[1])//所求范围，包含子树    {        ans += tree[o].sum;        return;    }    if(xl <= tree[o].d[0] && xr >= tree[o].d[0] && yl <= tree[o].d[1] && yr >= tree[o].d[1])//部分包含        ans += tree[o].v;    if(tree[o].l) query(tree[o].l);    if(tree[o].r) query(tree[o].r);}int read(){    int t;    scanf("%d", &t);    return ans^t;}int main(){    int n;    scanf("%d", &n);    int pos = 1, op;    int root = 0;    ans = 0;    while(true)    {        scanf("%d", &op);        if(op == 1)        {            for(int i = 0; i < DIM; i++)                tree[pos].d[i] = read();            tree[pos].v = read();            tree[pos].maintain();            Insert(root, pos, 0);            pos++;            if(pos % 5000 == 0)                root = build(1, pos-1, 0);        }        else if(op == 2)        {            xl = read();            yl = read();            xr = read();            yr = read();            ans = 0;            query(root);            printf("%d\n", ans);        }        else break;    }    return 0;}