(LeetCode 934)最短的橋 [簡單DFS + BFS]
阿新 • • 發佈:2019-01-01
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; } };