LeetCode 1568. 使陆地分离的最少天数

LeetCode 1568. 使陆地分离的最少天数

给你一个由若干 0 和 1 组成的二维网格 grid ，其中 0 表示水，而 1 表示陆地。岛屿由水平方向或竖直方向上相邻的 1 （陆地）连接形成。

如果 恰好只有一座岛屿 ，则认为陆地是 连通的 ；否则，陆地就是 分离的 。

一天内，可以将任何单个陆地单元（1）更改为水单元（0）。 

返回使陆地分离的最少天数。

class Solution {
public:
    int n,m;
    int dx[4]={0,0,1,-1},dy[4]={1,-1,0,0};
    vector<vector<bool>> visit;
    vector<vector<int>> g;
    void dfs(int a,int b){
        visit[a][b]=true;
        for(int i=0;i<4;i++){
            int x=a+dx[i];
            int y=b+dy[i];
            if(x<n&&x>=0&&y<m&&y>=0&&g[x][y]==1&&!visit[x][y]){
                dfs(x,y);
            }
        }
    }
    bool check(){
        int cnt=0;
        visit=vector<vector<bool> >(n,vector<bool>(m));
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(!visit[i][j]&&g[i][j]==1){
                    cnt++;
                    dfs(i,j);
                }
            }
        }
        return cnt>1||cnt==0;
    }
    int minDays(vector<vector<int>>& grid) {
        n=grid.size();
        m=grid[0].size();
        g=grid;
        if(check()) return 0;
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(g[i][j]==1){
                    g[i][j]=0;
                    if(check()) return 1;
                    g[i][j]=1;
                }
            }
        }
        return 2;
    }
};