题意:
公墓可以看成一块N×M 的矩形,矩形的每个格点,要么种着一棵常青树,要么是一块墓地。一块墓地的虔诚度是指以这块墓地为中心的十字架的数目。一个十字架可以看成中间是墓地,墓地的正上、正下、正左、正右都有恰好k 棵常青树。小W 希望知道他所管理的这片公墓中所有墓地的虔诚度总和是多少。
题解:
再一次看错题
认真想了一会几乎毫无头绪,反正暴力感觉挺好写,先离散化(因为对于一个有用的墓地,其上下左右一定是有树的,所以这个点的坐标还在离散值里面)
然后找出每一墓地上下左右分别有多少棵树,组合数乘一下。
orz题解,发现其实不用枚举墓地,因为墓地一定在两棵相邻的树中间,所以将数排序后从上往下扫,用树状数组维护下组合数乘积的前缀和就好了。
代码还是挺好写的。
code:
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define llg unsigned int
using namespace std;
struct node{int x,y,num;}s[100010];
int n,m,W,num=0,X[100010],k,wx[100010],N;
int l[100010],r[100010],d[100010],u[100010];
llg c[100010],Cc[100010][11],ans=0;
bool cmp1(node a,node b)
{
if(a.x==b.x) return a.y<b.y;
return a.x<b.x;
}
bool cmp2(node a,node b)
{
if(a.y==b.y) return a.x<b.x;
return a.y<b.y;
}
void pre()
{
sort(s+1,s+W+1,cmp1);
for(int i=1;i<=W;i++)
{
int j=i,now=0;
while(j<W&&s[j+1].x==s[j].x) j++,d[s[j].num]=++now;
while(i<=j) u[s[i].num]=now--,i++;
i--;
N=max(N,d[s[j].num]);
}
sort(s+1,s+W+1,cmp2);
for(int i=1;i<=W;i++)
{
int j=i,now=0;
while(s[j+1].y==s[j].y) j++,l[s[j].num]=++now;
while(i<=j) r[s[i].num]=now--,i++;
i--;
N=max(N,l[s[j].num]);
}
for(int i=0;i<=N;i++) Cc[i][0]=Cc[i][i]=1;
for(int i=2;i<=N;i++)
for(int j=1;j<=k;j++)
Cc[i][j]=Cc[i-1][j-1]+Cc[i-1][j];
}
inline llg C(int x){return Cc[x][k];}
void change(int x,llg y){while(x<=num) c[x]+=y,x+=x&(-x);}
llg get(int x){llg re=0;while(x) re+=c[x],x-=x&(-x);return re;}
int main()
{
scanf("%d %d %d",&n+1,&m+1,&W);
for(int i=1;i<=W;i++)
{
int x,y;scanf("%d %d",&x,&y);
X[++num]=s[i].x=x+1;s[i].y=y+1;s[i].num=i;
}
scanf("%d",&k);
sort(X+1,X+num+1);
num=unique(X+1,X+num+1)-X-1;
pre();
for(int i=1;i<=W;i++) wx[i]=lower_bound(X+1,X+num+1,s[i].x)-X;
for(int i=1,j;i<=W;i=j+1)
{
j=i;
change(wx[i],C(d[s[i].num]+1)*C(u[s[i].num])-C(d[s[i].num])*C(u[s[i].num]+1));
while(j<W&&s[j+1].y==s[j].y)
{
j++;
ans+=C(l[s[j-1].num]+1)*C(r[s[j].num]+1)*(get(wx[j]-1)-get(wx[j-1]));
change(wx[j],C(d[s[j].num]+1)*C(u[s[j].num])-C(d[s[j].num])*C(u[s[j].num]+1));
}
}
printf("%u",ans&2147483647);
}