Description
维护一个W*W的矩阵,初始值均为S.每次操作可以增加某格子的权值,或询问某子矩阵的总权值.修改操作数M<=160000,询问数Q<=10000,W<=2000000.
Input
第一行两个整数,S,W;其中S为矩阵初始值;W为矩阵大小
接下来每行为一下三种输入之一(不包含引号):
"1 x y a"
"2 x1 y1 x2 y2"
"3"
输入1:你需要把(x,y)(第x行第y列)的格子权值增加a
输入2:你需要求出以左上角为(x1,y1),右下角为(x2,y2)的矩阵内所有格子的权值和,并输出
输入3:表示输入结束
Output
对于每个输入2,输出一行,即输入2的答案
Sample Input
0 4
1 2 3 3
2 1 1 3 3
1 2 2 2
2 2 2 3 4
3
1 2 3 3
2 1 1 3 3
1 2 2 2
2 2 2 3 4
3
Sample Output
3
5
5
HINT
保证答案不会超过int范围
首先我们将所有的操作都储存下来,至于查询操作由于是查询一个二维范围内的值,所以可以把它差分成四个不同的查询操作,最后记录对于答案造成的贡献。然后按照x轴排序,开始分治,根据cdq分治的思想,简单来说就是分治处理两块,然后先处理左边的操作,然后这些操作对右边来说,利用左边处理出来的东西消除左边对右边的影响,然后继续分治。和整体二分类似入门的话比较难理解。
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#define maxn 2100020
using namespace std;
int cnt,s,w,ans[maxn],c[maxn*4];
struct node{
int id,x,y,d,pos,k;
bool operator<(const node& bb)const{
if(x==bb.x&&y==bb.y)return id<bb.id;
return x==bb.x?y<bb.y:x<bb.x;
}
}q[maxn],t[maxn];
void update(int x,int add){while(x<=w){c[x]+=add;x+=x&(-x);}}
int query(int x){
int ans=0;
while(x>0){ans+=c[x];x-=x&(-x);}
return ans;
}
void solve(int l,int r){
if(l==r)return;
int mid=l+r>>1;
int j=l,k=mid+1;
for(int i=l;i<=r;i++){
if(q[i].id<=mid&&q[i].pos==1)update(q[i].y,q[i].d);
else if(q[i].id>mid&&q[i].pos==2)ans[q[i].k]+=query(q[i].y)*q[i].d;
}
for(int i=l;i<=r;i++)
if(q[i].id<=mid&&q[i].pos==1)update(q[i].y,-q[i].d);
for(int i=l;i<=r;i++)
if(q[i].id <= mid)t[j++]=q[i];else t[k++]=q[i];
for(int i=l;i<=r;i++)q[i]=t[i];
solve(l,mid),solve(mid+1,r);
}
int main(){
int m,tot=0,a,b,c,d;
scanf("%d%d",&s,&w);
while(scanf("%d",&m)&&m!=3){
if(m==1){
scanf("%d%d%d",&a,&b,&c);
q[++cnt]=(node){cnt,a,b,c,1,0};
}else{
tot++;
scanf("%d%d%d%d",&a,&b,&c,&d);
q[++cnt]=(node){cnt,c,d,1,2,tot};
q[++cnt]=(node){cnt,a-1,b-1,1,2,tot};
q[++cnt]=(node){cnt,c,b-1,-1,2,tot};
q[++cnt]=(node){cnt,a-1,d,-1,2,tot};
}
}
sort(q+1,q+1+cnt);
solve(1,cnt);
for(int i=1;i<=tot;i++)printf("%d\n",ans[i]);
return 0;
}