cf723 Round #375 Div2-D【Floodfill+搜索】

Date：2022.01.19
题意：给定n*m的岛，’.‘表示此处为水，连着的水为湖。连着边界的所有水为海，不算入湖。给定k，要求保留k个湖，至少需要填多少格像’.‘这样水，并输出填湖之后的岛。

思路：先预处理所有连着海的点，都不算入湖。之后找到每个未被访问的点，dfs1找到它所在的连通块标记并为访问过。之后每个连通块取一个代表，bfs版Floodfill找到每个连通块的大小，加入vector中排序，保留最大的k个，其余算累加即为需要填补的’.'的数量；再找到需要的连通块，dfs2填上每个坑输出。（直接莽的确实太乱了）
代码如下：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int N = 55;
typedef long long LL;
typedef pair<int, int> PII;
LL n,m,k;
LL a[N][N];
bool st[N][N];
char c[N][N];
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
void dfs1(LL x,LL y)
{
    if(c[x][y]=='.') st[x][y]=true;
    for(int i=0;i<4;i++)
    {
        LL ix=x+dx[i],iy=y+dy[i];
        if(ix<=0||iy<=0||ix>n||iy>m||c[ix][iy]=='*'||st[ix][iy]) 
            continue;
        st[ix][iy]=true;
        dfs1(ix,iy);
    }
}
void dfs2(LL x,LL y)
{
    if(c[x][y]=='.') c[x][y]='*';
    for(int i=0;i<4;i++)
    {
        LL ix=x+dx[i],iy=y+dy[i];
        if(ix<=0||iy<=0||ix>n||iy>m||c[ix][iy]=='*')
            continue;
        c[ix][iy]='*';
        dfs2(ix,iy);
    }
}
LL bfs(LL x,LL y)
{
    queue<PII>q;
    LL area=0;
    q.push({x,y});
    st[x][y]=true;
    while(q.size())
    {
        PII t=q.front();q.pop();
        area++;
        for(int i=0;i<4;i++)
        {
            LL ix=t.first+dx[i],iy=t.second+dy[i];
            if(ix<=0||iy<=0||ix>n||iy>m||c[ix][iy]=='*'||st[ix][iy])
                continue;
            q.push({ix,iy});
            st[ix][iy]=true;
        }
    }
    return area;
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    cin>>n>>m>>k;
    vector<PII>v;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            cin>>c[i][j];
            if((i==1||i==n)&&c[i][j]=='.') v.push_back({i,j});
            if((j==1||j==m)&&c[i][j]=='.') v.push_back({i,j});
        }
    for(int i=0;i<v.size();i++)
    {
        LL x=v[i].first,y=v[i].second;
        if(st[x][y]) continue;
        dfs1(x,y);
    }
    v.clear();
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            if(!st[i][j]&&c[i][j]=='.')
                v.push_back({i,j});
    LL maxx=0;queue<LL>qq;
    vector<LL>v1;vector<PII>v2;
    for(int i=0;i<v.size();i++)
    {
        LL x=v[i].first,y=v[i].second;
        if(!st[x][y] && c[x][y]=='.')
        {
            LL ans=bfs(x,y);a[x][y]=ans;
            v1.push_back(ans);v2.push_back({x,y});
        }
    }
    LL num=0,res=0;
    sort(v1.begin(),v1.end());
    for(int i=0;i<v1.size();i++) 
    {
        if(i<v1.size()-k) res+=v1[i];
        qq.push(v1[i]);
    }
    LL ss=qq.size();//几个湖
    memset(st,false,sizeof st);
    while(qq.size())
    {
        LL t=qq.front();qq.pop();
        bool flag=false;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=m;j++)
                if(a[i][j]==t && !st[i][j])
                {
                    dfs2(i,j);st[i][j]=true;flag=true;num++;
                    break;
                }
            if(flag || num==ss-k) break;
        }
    }
    cout<<res<<endl;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++) cout<<c[i][j];
        cout<<endl;
    }
    return 0;
}