(LeetCode 934)最短的桥 [简单DFS + BFS]

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/STILLxjy/article/details/85341733

934. 最短的桥
在给定的二维二进制数组 A 中,存在两座岛。(岛是由四面相连的 1 形成的一个最大组。)

现在,我们可以将 0 变为 1,以使两座岛连接起来,变成一座岛。

返回必须翻转的 0 的最小数目。(可以保证答案至少是 1。)

示例 1:

输入:[[0,1],[1,0]]
输出:1
示例 2:

输入:[[0,1,0],[0,0,0],[0,0,1]]
输出:2
示例 3:

输入:[[1,1,1,1,1],[1,0,0,0,1],[1,0,1,0,1],[1,0,0,0,1],[1,1,1,1,1]]
输出:1

提示:
1 <= A.length = A[0].length <= 100
A[i][j] == 0 或 A[i][j] == 1

分析:
(1)首先找到某一个岛屿上的一点(x,y),利用点(x,y)和岛屿的连通性,DFS找出这个岛屿的所有边界,即距离该岛屿距离为1的所有‘0’的点,将这些边界点的坐标存入将队列中。

(2)根据队列中的信息,使用BFS不断进行扩展,直到找到另外一个岛屿(即到达某个未访问的‘1’)。

AC代码:

class Solution {
public:
    bool vis[110][110];
    int dx[4] = {-1,0,1,0};
    int dy[4] = {0,1,0,-1};
    int m, n;
    
    bool check(int x,int y)
    {
        return x>=0 && x< m && y>=0 && y<n;
    }
    
    void dfs(int x, int y,vector<vector<int>>& A, queue<int>& q)
    {
        vis[x][y] = 1;
     
        for(int k=0;k<4;k++)
        {
            int xx = x + dx[k];
            int yy = y + dy[k];
            if(check(xx,yy) && vis[xx][yy] == 0 && A[xx][yy] == 1)
            {
                dfs(xx,yy,A,q);
            }
            else if(check(xx,yy) && vis[xx][yy] == 0 && A[xx][yy] == 0)
            {
                q.push(xx);
                q.push(yy);
                q.push(1);
            }
        }
    }
    
    int shortestBridge(vector<vector<int>>& A) {
        queue<int> q;
        m = A.size();
        n = A[0].size();
        memset(vis,0,sizeof(vis));
        int x, y;
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                if(A[i][j] == 1)
                {
                    x = i;
                    y = j;
                    break;
                } 
            }
        }
        dfs(x,y,A,q);
        
        int len = 0;
        while(!q.empty())
        {
            x = q.front();
            q.pop();
            y = q.front();
            q.pop();
            len = q.front();
            q.pop();
            
            if(A[x][y] == 1)
            {
                return len - 1;
            }
            
            for(int k=0;k<4;k++)
            {
                int xx = x + dx[k];
                int yy = y + dy[k];
                if(check(xx,yy) && vis[xx][yy] == 0)
                {
                    vis[xx][yy] = 1;
                    q.push(xx);
                    q.push(yy);
                    q.push(len + 1);
                }
            }
            
        }
        return -1;
    }
};

猜你喜欢

转载自blog.csdn.net/STILLxjy/article/details/85341733