二维线段树之树套树

//poj1195 二维线段树之树套树
// 先确定横坐标所在的区间并记录该结点的编号p，然后再确定纵坐标所在的区间并记录该结点的编号cur，则tree[cur][p]为目标区间。
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <queue>
#include <stack>
#include <deque>
#include <map>
#include <iostream>
using namespace std;
typedef long long LL;
const double pi = acos(-1.0);
const double e = exp(1);
//const int MAXN =2e5+10;
const int N = 1055;

//const int maxn = 1024 * 1024 * 1.5;

int n;
int sum[N * 3][N * 3];

void updatey(int y, int l, int r, int cur, int p, int op, int val)
// 纵坐标目标值，待找区间的左端点，待找区间的右端点，y中树的结点的下标，已找到的横坐标中结点的下标，值更新的方式，需要更新的值
{
    if(l == r)
    {
        if(op == 1)  //根据题目要求，对目标值直接修改
        {
            sum[p][cur] += val;
        }
        else   // 回溯过程中，维护树的结构
        {
            sum[p][cur] = sum[p << 1][cur] + sum[p << 1 | 1][cur];
        }
        return ;
    }
    int mid = (l + r) >> 1;
    if(y <= mid)   //当前结点的左孩子区间需要找
    {
        updatey(y, l, mid, cur << 1, p, op, val);
    }
    else  // 当前结点的右孩子区间需要找
    {
        updatey(y, mid + 1, r, cur << 1 | 1, p, op, val);
    }
    sum[p][cur] = sum[p][cur << 1] + sum[p][cur << 1 | 1];

}

void updatex(int x, int y, int l, int r, int cur, int val)
{
    if(l == r)
    {
        updatey(y, 1, n, 1, cur, 1, val);   //锁定了横坐标所在的结点cur，再去寻找纵坐标所在的结点。
        return ; 
    }
    int mid = (l + r) >> 1;
    if(x <= mid)
    {
        updatex(x, y, l, mid, cur << 1, val);
    }
    else
    {
        updatex(x, y, mid + 1, r, cur << 1 | 1, val);
    }
    updatey(y, 1, n, 1, cur, 2, val);
}

int queryy(int ly, int ry, int l, int r, int cur, int p)
{
    if(ly <= l && ry >= r)
    {
        return sum[p][cur];
    }
    int res = 0;
    int mid = (l + r) >> 1;
    if(ly <= mid)
    {
        res += queryy(ly, ry, l, mid, cur << 1, p);
    }
    if(ry > mid)
    {
        res += queryy(ly, ry, mid + 1, r, cur << 1 | 1, p);
    }
    return res;
}

int queryx(int lx, int rx, int ly, int ry, int l, int r, int cur)
{
    int res = 0;
    if(lx <= l && rx >= r)
    {
        res = queryy(ly, ry, 1, n, 1, cur);
        return res;
    }
    int mid = (l + r) >> 1;
    if(lx <= mid)
    {
        res += queryx(lx, rx, ly, ry, l, mid, cur << 1);
    }
    if(rx > mid)
    {
        res += queryx(lx, rx, ly, ry, mid + 1, r, cur << 1 | 1);
    }
    return res;
}


int main()
{
    int i, j, t;
    int x1, y1, x2, y2, c;

    while(scanf("%d",&t) != EOF)
    {
        if(t == 3)
            break;
        if(t == 0)
        {
            scanf("%d",&n);
            n ++;
            memset(sum, 0, sizeof(sum));
        }
        else if(t == 1)
        {
            scanf("%d%d%d", &x1, &y1, &c);
            x1 ++, y1 ++;
            updatex(x1, y1, 1, n, 1, c);
        }
        else if(t == 2)
        {
            scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
            x1++, y1++, x2++, y2++;
            printf("%d
", queryx(x1, x2, y1, y2, 1, n, 1));
        }
    }

    return 0;
}