题目解析: 本题考查DFS/BFS的联合应用,首先我们搜索出一个岛屿之后通过BFS找到两个岛屿之间的最近距离即可。
重点: 这里的考点在于BFS层数的设置,我们可以通过记录队列中节点个数来达到效果。一次只搜索一层的节点。
while(!q.empty())
{
level++;
int q_size = q.size();
while(q_size--)
{
auto [r, c] = q.front();
q.pop();
}
}
vector<vector<int>> direction{{0,1},{1,0},{0,-1},{-1,0}};
class Solution {
public:
int m,n;
void DFS(int i,int j, vector<vector<int>>& A, queue<pair<int,int>>&q)
{
q.push(make_pair(i,j));
A[i][j]++;
for(auto it:direction)
{
if(0<=i+it[0]&&i+it[0]<m&&0<=j+it[1]&&j+it[1]<n&&A[i+it[0]][j+it[1]]==1)
{
DFS(i+it[0],j+it[1],A,q);
}
}
}
int shortestBridge(vector<vector<int>>& A) {
bool flag = true;
m = A.size();
n = A[0].size();
queue<pair<int,int>> q;
for(int i = 0;i<A.size()&&flag;i++)
{
for(int j = 0;j<A[0].size()&&flag;j++)
{
if(A[i][j]==1)
{
DFS(i,j,A,q);
flag=false;
}
}
}
int level = 0;
while(!q.empty())
{
level++;
int q_size = q.size();
while(q_size--)
{
auto [r, c] = q.front();
q.pop();
for(auto it:direction)
{
if(0<=r+it[0]&&r+it[0]<m&&0<=c+it[1]&&c+it[1]<n&&A[r+it[0]][c+it[1]]!=2)
{
if(A[r+it[0]][c+it[1]]==0)
{
A[r+it[0]][c+it[1]]=2;
q.push(make_pair(r+it[0],c+it[1]));
}
else
{
return level-1;
}
}
}
}
}
return level;
}
};