Codeforces Round #356 (Div. 2)E. Bear and Square Grid

链接：http://codeforces.com/contest/680/problem/E

题意：给定一个n*n的地图和一个k，地图上有空地和障碍物两种。你能进行之多一次操作，操作：将一块k*k的区域内的所有障碍物清除。问操作完之后最大的一块空地的连通块能有多大。

分析：因为n只有500，我们枚举操作打掉的那一个k*k的右下角(i,j)。那么我们只需要用4*k的复杂度去将它周围能连通的块都统计起来，但是在k*k的块里面我们不能将里面能与外面连通的块重复计算，我们在统计周围的连通块时就统计过了。那么还需要加的是整个连通块都在k*k里面和k*k里面的障碍物的个数。障碍物容易算，容斥一下就行了。就只有算整个都在里面的连通块的大小了，我们可以反向想一想，对于一个联通块p，它会整个在哪些点为右下角的那个k*k里面呢？这些点是一个矩形，我们在4个顶点上做点标记处理下前缀和即可。详见代码。O(n^2*k)

代码：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=510;
const int MAX=1000000100;
const int mod=100000000;
const int MOD1=1000000007;
const int MOD2=1000000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=998244353;
const int INF=1000000010;
const double pi=acos(-1.0);
typedef double db;
typedef unsigned long long ull;
char s[N];
int n,a[2*N][2*N],b[N][N];
int g=0,q[N*N],f[N*N],xm[N*N],ym[N*N],xx[N*N],yx[N*N];
void dfs(int x,int y) {
    b[x][y]=g;f[g]++;
    xm[g]=min(xm[g],x);xx[g]=max(xx[g],x);
    ym[g]=min(ym[g],y);yx[g]=max(yx[g],y);
    if (x>1&&b[x-1][y]==-1) dfs(x-1,y);
    if (x<n&&b[x+1][y]==-1) dfs(x+1,y);
    if (y>1&&b[x][y-1]==-1) dfs(x,y-1);
    if (y<n&&b[x][y+1]==-1) dfs(x,y+1);
}
int main()
{
    int i,j,h,k,tot,ans=0;
    scanf("%d%d", &n, &k);
    memset(a,0,sizeof(a));
    for (i=1;i<=n;i++) {
        scanf("%s", s);
        for (j=1;j<=n;j++)
        if (s[j-1]=='.') b[i][j]=-1;
        else {
            b[i][j]=0;
            a[i][j]+=1;a[i][j+k]+=-1;
            a[i+k][j]+=-1;a[i+k][j+k]+=1;
        }
    }
    memset(f,0,sizeof(f));
    memset(xx,0,sizeof(xx));
    memset(yx,0,sizeof(yx));
    memset(xm,0x7f,sizeof(xm));
    memset(ym,0x7f,sizeof(ym));
    for (i=1;i<=n;i++)
        for (j=1;j<=n;j++)
        if (b[i][j]==-1) {
            g++,dfs(i,j);
            if (xx[g]-xm[g]+1<=k&&yx[g]-ym[g]+1<=k) {
                a[xx[g]][yx[g]]+=f[g];a[xm[g]+k][ym[g]+k]+=f[g];
                a[xx[g]][ym[g]+k]-=f[g];a[xm[g]+k][yx[g]]-=f[g];
            }
        }
    for (i=1;i<=n;i++)
        for (j=1;j<=n;j++) a[i][j]+=a[i][j-1]+a[i-1][j]-a[i-1][j-1];
    memset(q,0,sizeof(q));
    for (i=k;i<=n;i++)
        for (j=k;j<=n;j++) {
            tot=0;
            for (h=j-k+1;h<=j;h++)
            if (!q[b[i-k][h]]) q[b[i-k][h]]=1,tot+=f[b[i-k][h]];
            for (h=j-k+1;h<=j;h++)
            if (!q[b[i+1][h]]) q[b[i+1][h]]=1,tot+=f[b[i+1][h]];
            for (h=i-k+1;h<=i;h++)
            if (!q[b[h][j-k]]) q[b[h][j-k]]=1,tot+=f[b[h][j-k]];
            for (h=i-k+1;h<=i;h++)
            if (!q[b[h][j+1]]) q[b[h][j+1]]=1,tot+=f[b[h][j+1]];
            ans=max(ans,tot+a[i][j]);
            for (h=j-k+1;h<=j;h++) q[b[i-k][h]]=0;
            for (h=j-k+1;h<=j;h++) q[b[i+1][h]]=0;
            for (h=i-k+1;h<=i;h++) q[b[h][j-k]]=0;
            for (h=i-k+1;h<=i;h++) q[b[h][j+1]]=0;
        }
    printf("%d\n", ans);
    return 0;
}