二维线段树就是树套树，线段树套线段树。。。
#include<iostream> #include<cstdio> #include<cstring> #include<cstdlib> #include<algorithm> #define REP(i,a,b) for(int i=a;i<=b;i++) #define MS0(a) memset(a,0,sizeof(a)) #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 using namespace std; typedef long long ll; const int maxn=1500; const int INF=1e9+10; int n; int op; int x,y,z; int xt1,yt1,xt2,yt2; struct NodeY { int l,r; int cnt; }; struct NodeX { int l,r; int cnt; NodeY ty[maxn<<2]; void up(int rt) { ty[rt].cnt=ty[rt<<1].cnt+ty[rt<<1|1].cnt; } void build(int l,int r,int rt) { ty[rt].l=l;ty[rt].r=r; ty[rt].cnt=0; if(l==r) return; int m=(l+r)>>1; build(lson); build(rson); up(rt); } void update(int p,int c,int l,int r,int rt) { if(l==r){ ty[rt].cnt+=c; return; } int m=(l+r)>>1; if(p<=m) update(p,c,lson); else update(p,c,rson); up(rt); } int query(int L,int R,int l,int r,int rt) { if(L<=l&&r<=R) return ty[rt].cnt; int m=(l+r)>>1; int res=0; if(L<=m) res+=query(L,R,lson); if(R>m) res+=query(L,R,rson); return res; } };NodeX tx[maxn<<2]; void up(int rt) { tx[rt].cnt=tx[rt<<1].cnt+tx[rt<<1|1].cnt; } void build(int l,int r,int rt) { tx[rt].l=l;tx[rt].r=r; tx[rt].cnt=0; tx[rt].build(1,n,1); if(l==r) return; int m=(l+r)>>1; build(lson); build(rson); up(rt); } void update(int x,int y,int c,int l,int r,int rt) { tx[rt].update(y,c,1,n,1); if(l==r) return; int m=(l+r)>>1; if(x<=m) update(x,y,c,lson); else update(x,y,c,rson); up(rt); } int query(int xL,int xR,int yL,int yR,int l,int r,int rt) { if(xL<=l&&r<=xR) return tx[rt].query(yL,yR,1,n,1); int m=(l+r)>>1; int res=0; if(xL<=m) res+=query(xL,xR,yL,yR,lson); if(xR>m) res+=query(xL,xR,yL,yR,rson); return res; } int main() { freopen("in.txt","r",stdin); scanf("%d%d",&op,&n); build(1,n,1); while(~scanf("%d",&op)&&op!=3){ if(op==1){ scanf("%d%d%d",&x,&y,&z); x++;y++; update(x,y,z,1,n,1); } else{ scanf("%d%d%d%d",&xt1,&yt1,&xt2,&yt2); xt1++;yt1++;xt2++;yt2++; printf("%d ",query(xt1,xt2,yt1,yt2,1,n,1)); } } return 0; }
-----更新
上面那个是利用了求和以及更新增减的特殊性，下面这个才是更实用的普遍做法。
#include<iostream> #include<cstdio> #include<cstring> #include<cstdlib> #include<algorithm> #define REP(i,a,b) for(int i=a;i<=b;i++) #define MS0(a) memset(a,0,sizeof(a)) #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 using namespace std; typedef long long ll; const int maxn=1500; const int INF=1e9+10; int op,n; int x,y,a; int xt,yt; int tr[maxn<<2][maxn<<2]; void upy(int rt,int rtx) { tr[rtx][rt]=tr[rtx][rt<<1]+tr[rtx][rt<<1|1]; } void buildy(int l,int r,int rt,int rtx) { if(l==r){ tr[rtx][rt]=0; return; } int m=(l+r)>>1; buildy(lson,rtx); buildy(rson,rtx); upy(rt,rtx); } void updatey(int p,int c,int l,int r,int rt,int rtx) { if(l==r){ tr[rtx][rt]+=c; return; } int m=(l+r)>>1; if(p<=m) updatey(p,c,lson,rtx); else updatey(p,c,rson,rtx); upy(rt,rtx); } int queryy(int L,int R,int l,int r,int rt,int rtx) { if(L<=l&&r<=R) return tr[rtx][rt]; int m=(l+r)>>1; int res=0; if(L<=m) res+=queryy(L,R,lson,rtx); if(R>m) res+=queryy(L,R,rson,rtx); return res; } void buildx(int l,int r,int rt) { buildy(1,n,1,rt); if(l==r) return; int m=(l+r)>>1; buildx(lson); buildx(rson); } void upx(int p,int rtx,int l,int r,int rt) { tr[rtx][rt]=tr[rtx<<1][rt]+tr[rtx<<1|1][rt]; if(l==r) return; int m=(l+r)>>1; if(p<=m) upx(p,rtx,lson); else upx(p,rtx,rson); } void updatex(int x,int y,int c,int l,int r,int rt) { if(l==r){ updatey(y,c,1,n,1,rt); return; } int m=(l+r)>>1; if(x<=m) updatex(x,y,c,lson); else updatex(x,y,c,rson); upx(y,rt,1,n,1); } int queryx(int xL,int xR,int yL,int yR,int l,int r,int rt) { if(xL<=l&&r<=xR) return queryy(yL,yR,1,n,1,rt); int m=(l+r)>>1; int res=0; if(xL<=m) res+=queryx(xL,xR,yL,yR,lson); if(xR>m) res+=queryx(xL,xR,yL,yR,rson); return res; } int main() { freopen("in.txt","r",stdin); scanf("%d%d",&op,&n); buildx(1,n,1); while(~scanf("%d",&op)&&op!=3){ if(op==1){ scanf("%d%d%d",&x,&y,&a); x++;y++; updatex(x,y,a,1,n,1); } else{ scanf("%d%d%d%d",&x,&y,&xt,&yt); x++;y++;xt++;yt++; printf("%d ",queryx(x,xt,y,yt,1,n,1)); } } return 0; }