poj1195

链接:点击打开链接

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

poj1195_第1张图片

代码:
#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;
}

你可能感兴趣的:(poj1195)