Atcoder 500 Checker 思维+二维前缀和

时间:2021-05-02 23:23:36
题意:一个无限大的二维平面 只能涂两种颜色,现在有n个要求(x,y,0/1)表示(x,y)上的点颜色要涂为黑/白.
给出n,k.在平面上的的任意一个联通分量大小都为k*k的条件下,最多能满足多少个要求?
n<=1e5, 0<=x,y<=1e9,k<=1e3.


枚举一个点作为某个颜色的左上角 因为任意联通分量的大小都为k*k 所以现在每个点的颜色都固定了
在遍历条件看有多少个满足 O(n*k*k).

因为(x,y)和(x+2k,y),(x,y+2k)的颜色一定是相同的.把所有点都转换为(x%2k,y%2k).
(x,y)和(x,y+k)的颜色一定是相反的 若(x,y)要求为白色 都等价为(x,y+k)黑色

枚举连通分量的左上角 用二维前缀和求出有多少个点若在里面即可O(K^2).

#include <bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int N=2e5+5,M=6008;
int n,k,x,y,m;
string c;
int f[M][M];
int calc(int x1,int y1)
{
int x2=x1+k-1,y2=y1+k-1;
x1=max(1,x1),y1=max(1,y1);
x2=min(x2,m),y2=min(y2,m);
if(x2<x1||y2<y1)
return 0;
return f[x2][y2]-f[x1-1][y2]-f[x2][y1-1]+f[x1-1][y1-1];
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cin>>n>>k;
m=2*k;
memset(f,0,sizeof(f));
rep(i,1,n)
{
cin>>x>>y>>c;
if(c=="W")
y+=k;
x%=m,y%=m;
x++,y++;
f[x][y]++;
}
for(int i=1;i<=2*m;i++)
for(int j=1;j<=2*m;j++)
f[i][j]=f[i][j]+f[i][j-1]+f[i-1][j]-f[i-1][j-1];
int res=0;
for(int i=1;i<=m;i++)
{
for(int j=1;j<=m;j++)
{
int cnt=calc(i,j);
cnt+=calc(i-k,j-k)+calc(i-k,j+k)+calc(i+k,j-k)+calc(i+k,j+k);
cnt+=calc(i-m,j)+calc(i+m,j)+calc(i,j-m)+calc(i,j+m);
res=max(res,cnt);
}
}
cout<<res<<endl;
return 0;
}