1. 程式人生 > >(LeetCode 934)最短的橋 [簡單DFS + BFS]

(LeetCode 934)最短的橋 [簡單DFS + BFS]

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;
    }
};