codeforces679C Bear and Square Grid（dfs优化）

Posted 2020-07-28 恶devil魔

题意：

给你n*n的矩阵（n<=500），矩阵内有x和.，然后给你一个k

你可以把一个k*k的矩阵内全部变成.

问你最多有多少个.可以联通

思路：

n^2枚举炸的位置，先预处理联通块和区间.的和

每次向右枚举只需要删掉左边一列，加上右边一列

每次枚举的区间是k*k然后扩展一圈（(k+2)*(k+2)）去掉四个角

这些点所在的联通块都是可以连通的

每次把这些联通块加起来，加上k*k的矩阵，减去这个矩阵内原本的.数量（预处理过）

O（n^2*k）

/* ***********************************************
Author        :devil
Created Time  :2016/6/23 15:27:25
************************************************ */
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <stdlib.h>
using namespace std;
const int N=510;
char s[N][N];
int sum[N][N];
int mp[N*N];
int vis[N][N];
int cur,id,all,n,k;
int vsc[N*N];
void dfs(int x,int y)
{
    cur++;
    vis[x][y]=id;
    if(s[x-1][y]==‘.‘&&!vis[x-1][y]) dfs(x-1,y);
    if(s[x+1][y]==‘.‘&&!vis[x+1][y]) dfs(x+1,y);
    if(s[x][y-1]==‘.‘&&!vis[x][y-1]) dfs(x,y-1);
    if(s[x][y+1]==‘.‘&&!vis[x][y+1]) dfs(x,y+1);
}
void push(int x,int y)
{
    if(!vis[x][y]) return ;
    if(!vsc[vis[x][y]]) all+=mp[vis[x][y]];
    vsc[vis[x][y]]++;
}
void pop(int x,int y)
{
    if(!vis[x][y]) return ;
    vsc[vis[x][y]]--;
    if(!vsc[vis[x][y]]) all-=mp[vis[x][y]];
}
int cal(int x,int y)
{
    return sum[x][y]-sum[x-k][y]-sum[x][y-k]+sum[x-k][y-k];
}
int main()
{
    //freopen("in.txt","r",stdin);
    int ans=0;
    id=1;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
        scanf("%s",s[i]+1);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            sum[i][j]=sum[i][j-1]+sum[i-1][j]-sum[i-1][j-1];
            if(s[i][j]==‘.‘) sum[i][j]++;
        }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if(s[i][j]==‘.‘&&!vis[i][j])
            {
                cur=0;
                dfs(i,j);
                mp[id++]=cur;
            }
    for(int i=1;i<=n-k+1;i++)
    {
        all=0;
        memset(vsc,0,sizeof(vsc));
        for(int j=i-1;j<=i+k;j++)
            for(int l=1;l<=k;l++)
                push(j,l);
        for(int j=i;j<i+k;j++)
            push(j,k+1);
        ans=max(ans,all+k*k-cal(i+k-1,k));
        for(int j=1;j<=n-k;j++)
        {
            for(int l=i;l<i+k;l++)
            {
                pop(l,j-1);
                push(l,j+k+1);
            }
            pop(i-1,j);
            pop(i+k,j);
            push(i-1,j+k);
            push(i+k,j+k);
            ans=max(ans,all+k*k-cal(i+k-1,j+k));
        }
    }
    printf("%d\n",ans);
    return 0;
}