1176: [Balkan2007]Mokia
Time Limit: 30 Sec Memory Limit: 162 MBSubmit: 1838 Solved: 810
[ Submit][ Status][ Discuss]
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
1 2 3 3
2 1 1 3 3
1 2 2 2
2 2 2 3 4
3
Sample Output
5
HINT
保证答案不会超过int范围
解题思路:cdq分治,首先一个矩形的值可以转化为四个矩形的加减。
然后对于每个询问拆成四个点,然后可以分成三维偏序,然后就是cdq
分治。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
int s,w,cnt,wen;
struct ss
{
int pan,x,y,z,mul,zhi;
}q[210000];
int zhi[2100000];
int ans[200000];
inline int read()
{
char y; int x=0,f=1; y=getchar();
while (y<'0' || y>'9') {if (y=='-') f=-1; y=getchar();}
while (y>='0' && y<='9') {x=x*10+int(y)-48; y=getchar();}
return x*f;
}
bool cmp(ss xg,ss yg)
{
if (xg.z!=yg.z)return xg.z<yg.z;else
if (xg.x!=yg.x)return xg.x<yg.x;else
return xg.y<yg.y;
}
void change(int now,int sug)
{
while (now<=w)
{
zhi[now]+=sug; now+=now&-now;
}
}
int query(int now)
{
int sug=0;
while (now>0)
{
sug+=zhi[now]; now-=now&-now;
}
return sug;
}
bool cog(ss xg,ss yg)
{
if (xg.x!=yg.x) return xg.x<yg.x;else
return xg.y<yg.y;
}
void cdq(int l,int r)
{
if (l==r)return;
int mid=(l+r)/2;
cdq(l,mid); cdq(mid+1,r);
sort(q+l,q+mid+1,cog); sort(q+mid+1,q+r+1,cog);
int tail=l;
for (int i=mid+1;i<=r;++i)
if (q[i].pan!=0)
{
while (tail<mid+1 && q[i].x>=q[tail].x)
{
change(q[tail].y,q[tail].zhi);
++tail;
}
int opp=query(q[i].y);
ans[q[i].pan]+=opp*q[i].mul;
}
for (int i=l;i<=tail-1;++i)
{
change(q[i].y,-q[i].zhi);
}
}
int main()
{
s=read(); w=read(); cnt=0; wen=0; int pn,ogg=0;
while (scanf("%d",&pn) && pn!=3)
{
++ogg;
if (pn==1)
{
++cnt;
q[cnt].x=read(); q[cnt].y=read(); q[cnt].zhi=read(); q[cnt].z=ogg;
}else
{
++wen; int xg=read(); int yg=read(); int xt=read(); int yt=read();
if (xg!=1 && yg!=1) ++cnt,q[cnt].pan=wen,q[cnt].x=xg-1,q[cnt].y=yg-1,q[cnt].mul=1,q[cnt].z=ogg,ans[wen]+=s*(xg-1)*(yg-1);
++cnt; q[cnt].pan=wen; q[cnt].x=xt; q[cnt].y=yt; q[cnt].mul=1; ans[wen]+=q[cnt].x*q[cnt].y*s; q[cnt].z=ogg;
if (xg!=1) ++cnt,q[cnt].pan=wen,q[cnt].x=xg-1,q[cnt].y=yt,q[cnt].mul=-1,ans[wen]-=q[cnt].x*q[cnt].y*s,q[cnt].z=ogg;
if (yg!=1) ++cnt,q[cnt].pan=wen,q[cnt].x=xt,q[cnt].y=yg-1,q[cnt].mul=-1,ans[wen]-=q[cnt].x*q[cnt].y*s,q[cnt].z=ogg;
}
}
sort(q+1,q+cnt+1,cmp);
cdq(1,cnt);
for (int i=1;i<=wen;++i)
{
printf("%d\n",ans[i]);
}
}