poj1195

链接：点击打开链接

题意：有四种指令，第一种初始化一个N*N的矩阵为0，第二种将坐标(X,Y)处的值添加A，第三种给出一个子矩阵的左上角和右上角，求矩阵内值的和，第四种结束所有指令，详细见下图

代码：

#include <stdio.h>
#include <string.h>
#include <iostream>
using namespace std;
const int SIZE=1050;
int n,val,X1,Y1,X2,Y2,tmpx,tmpy;
int sum[SIZE*3][SIZE*3];                        //不能乘4会MLE
void updatey(int x,int l,int r,int rtx,int rty){
    int m;
    if(l==r){                                   //参数x代表这个节点是不是这颗树的树叶
        if(x!=-1)
        sum[rtx][rty]+=val;                     //如果是树叶则直接修改
        else                                     //不是的话则是这个节点的左节点和右节点
        sum[rtx][rty]=sum[rtx<<1][rty]+sum[rtx<<1|1][rty];
        return;                                 //相同位置的值的和，与线段树pushup有些
    }                                            //但不是一棵树上pushup
    m=(l+r)>>1;
    if(tmpy<=m)
    updatey(x,l,m,rtx,rty<<1);
    if(tmpy>m)
    updatey(x,m+1,r,rtx,rty<<1|1);              //继续更新Y这颗树
    sum[rtx][rty]=sum[rtx][rty<<1]+sum[rtx][rty<<1|1];
}
void updatex(int l,int r,int rtx){
    int m;
    if(l==r){
        updatey(l,1,n,rtx,1);                   //找到X区间后更新Y区间，这时参数x
        return;                                 //为1
    }
    m=(l+r)>>1;
    if(tmpx<=m)
    updatex(l,m,rtx<<1);
    if(tmpx>m)
    updatex(m+1,r,rtx<<1|1);
    updatey(-1,1,n,rtx,1);                      //不是树叶的节点，参数x传-1
}
long long query_y(int l,int r,int rtx,int rty){
    int m;
    long long ans;
    ans=0;
    if(Y1<=l&&r<=Y2)
    return sum[rtx][rty];
    m=(l+r)>>1;
    if(Y1<=m)
    ans+=query_y(l,m,rtx,rty<<1);
    if(Y2>m)
    ans+=query_y(m+1,r,rtx,rty<<1|1);
    return ans;
}
long long query_x(int l,int r,int rtx){
    int m;
    long long ans;
    ans=0;
    if(X1<=l&&r<=X2)
    return query_y(1,n,rtx,1);
    m=(l+r)>>1;
    if(X1<=m)
    ans+=query_x(l,m,rtx<<1);
    if(X2>m)
    ans+=query_x(m+1,r,rtx<<1|1);
    return ans;
}                                               //询问就是二维线段树基本询问方式，先找到X区间再找Y区间
int main(){                                     //二维线段树单点修改，区间求和
    int cur,num;
    while(scanf("%d%d",&cur,&n)!=EOF){
        memset(sum,0,sizeof(sum));              //初始全为0，省去了建树的过程
        while(scanf("%d",&num)){
            if(num==3)
            break;
            if(num==1){
                scanf("%d%d%d",&tmpx,&tmpy,&val);
                tmpx++,tmpy++;
                updatex(1,n,1);
            }
            else{
                scanf("%d%d%d%d",&X1,&Y1,&X2,&Y2);
                X1++,X2++,Y1++,Y2++;
                printf("%I64d\n",query_x(1,n,1));
            }
        }
    }
    return 0;
}