题目链接:http://v.qzyz.com/contest/291/problem/2
题目大意:有一个w*h的网格和n滴墨水,第 i 滴墨水的坐标为P(xi,yi),有一对参数(ai,bi)。对于一个网格C(x,y),这滴墨水对它的影响是max(0,ai-bi*dist(P,C)),其中dist(P,C)=max(|x-xi|,|y-yi|)。最终每个网格的受损值是所有墨水对它的影响。给你q个询问,问某个子网格图(x1,y1)~(x2,y2)的平均受损值,答案四舍五入到整数。
数据范围:w*h<=2 500 000,n,q<=200 000,1<=ai,bi<=10^9
题解:显然暴力不可行,考虑如何打标记。我们知道二维前缀和sum[x][y]=sum[x-1][y]+sum[x][y-1]-sum[x-1][y-1]+a[x][y],于是差分之后只需要在四个角做个标记。例如在(x1,y1)~(x2,y2)这部分网格都加上一个数x,那么我们只要在(x1,y1)和(x2+1,y2+1)加上x,在(x1,y2+1)和(x2+1,y1)减去x就可以了。
至于具体怎么标记,举个例子:如图,点P是一滴墨水,a=7,b=2,影响随颜色由浅到深。
首先我们先把最外层加上a mod b,然后转化为a mod b=0的情况。这样接下来做的标记全部都是b的倍数。于是我们不考虑最外层,那么这滴墨水对网格的影响如下:
可以发现只需要从最外圈开始逐步往里,每次把当前这块矩形区域加2就可以了。需要打的标记是这样的:
显然我们可以再做一次差分,分别记录从左上到右下与从右上到左下的标记。
到这里思路应该已经很清晰了,然而还没完。因为可能存在网格外的点受到影响。
呵呵呵。
见下图:
其中红色区域表示网格内的部分,由图可知我们还需记录从左到右以及从上到下的标记。
读者可以自己画图yy一下。
最后前缀和一发算出每个网格受到的影响值,再前缀和一发算出以某个点为右下角的矩形区域的影响值之和,然后询问的时候就可以O(1)算出结果啦。
时间复杂度:O(?*w*h)
(ps:wa了n发结果万万没想到学校OJ居然卡double的精度,我说怎么本地输出和OJ显示不一样,不知道一中的OJ会不会这样hhhh懒得去试了)
代码如下:
#include <algorithm>
#include <cstdio>
#define A(x,y) ((x>0)&&(y>0)&&(x<=w)&&(y<=h)?(x-1)*h+y:0)
using namespace std;
const int fx[4]={1,1,0,1},fy[4]={1,-1,1,0};
long long c[2500010],d[4][2500010];
int w,h,n,q;
void work(int op,int x,int y,int xx,int yy)
{
for (x+=xx,y+=yy;A(x,y)>0;x+=xx,y+=yy)
d[op][A(x,y)]+=d[op][A(x-xx,y-yy)];
}
int main()
{
scanf("%d%d\n%d\n",&w,&h,&n);
for (int i=1;i<=n;++i)
{
int x,y,a,b,topx,botx,lefy,rigy,x1,y1,x2,y2,z;
scanf("%d%d%d%d\n",&x,&y,&a,&b);
topx=x-a/b;botx=x+a/b;
lefy=y-a/b;rigy=y+a/b;
x1=max(1,topx);x2=min(w,botx);
y1=max(1,lefy);y2=min(h,rigy);
c[A(x1,y1)]+=a%b;c[A(x2+1,y2+1)]+=a%b;
c[A(x1,y2+1)]-=a%b;c[A(x2+1,y1)]-=a%b;
x1=max(1,topx+1);x2=min(w,botx);
y1=max(1,lefy+1);y2=min(h,rigy);
a=a/b-1;
if (a<0) continue;
//-----------------------------------------------------------------
z=min(x-x1,y-y1);
d[0][A(x-z,y-z)]+=b;
d[0][A(x+1+a+1,y+1+a+1)]-=b;
if (x1<x-z) d[3][A(x1,y1)]+=b,d[3][A(x-z,y1)]-=b;
if (y1<y-z) d[2][A(x1,y1)]+=b,d[2][A(x1,y-z)]-=b;
c[A(x1,y1)]+=1ll*(a-max(x-x1,y-y1))*b;
//-----------------------------------------------------------------
++y;
z=min(x-x1,y2-y);
d[1][A(x-z,y+z)]-=b;
d[1][A(x+1+a+1,y-1-a-1)]+=b;
if (y2>y+z) d[2][A(x1,y+z+1)]-=b,d[2][A(x1,y2+1)]+=b;
z=min(x2-x-1,y-1-y1);
if (x2>x+1+z) d[3][A(x+1+z+1,y1)]-=b,d[3][A(x2+1,y1)]+=b;
}
for (int i=1;i<=w;++i)
{
work(0,i,1,fx[0],fy[0]);
work(1,i,h,fx[1],fy[1]);
work(2,i,1,fx[2],fy[2]);
}
for (int i=1;i<=h;++i)
{
if (i>1) work(0,1,i,fx[0],fy[0]);
if (i<h) work(1,1,i,fx[1],fy[1]);
work(3,1,i,fx[3],fy[3]);
}
for (int i=1,j;i<=w*h;++i)
for (j=0;j<4;++j) c[i]+=d[j][i];
c[0]=0;
for (int i=1,j;i<=w;++i)
for (j=1;j<=h;++j) c[A(i,j)]+=c[A(i-1,j)]+c[A(i,j-1)]-c[A(i-1,j-1)];
for (int i=1,j;i<=w;++i)
for (j=1;j<=h;++j) c[A(i,j)]+=c[A(i-1,j)]+c[A(i,j-1)]-c[A(i-1,j-1)];
scanf("%d\n",&q);
for (int i=1,x1,y1,x2,y2;i<=q;++i)
{
scanf("%d%d%d%d\n",&x1,&y1,&x2,&y2);--x1;--y1;
long long sum=(c[A(x2,y2)]-c[A(x1,y2)]-c[A(x2,y1)]+c[A(x1,y1)]),s=(x2-x1)*(y2-y1);
long long ans=sum/s;
if (sum%s>=(s+1)/2) ans++;
printf("%lld\n",ans);
}
return 0;
}