题目描述:
给你一个大小为 n x n 的二元矩阵 grid ,其中 1 表示陆地,0 表示水域。
岛 是由四面相连的 1 形成的一个最大组,即不会与非组内的任何其他 1 相连。grid 中 恰好存在两座岛 。
你可以将任意数量的 0 变为 1 ,以使两座岛连接起来,变成 一座岛 。
返回必须翻转的 0 的最小数目。
题目解析:
先通过一个infect函数将一座岛的值修改为2,然后遍历整个矩阵,对值为2的点进行计算,这里有一个优化技巧,就是对于整个岛的轮廓来进行计算,因为要计算两个岛的最短距离,那么最短距离的两个点肯定分别都是两个岛的轮廓点,这样可以简化计算,然后计算值为2的轮廓点到每一个值为1的轮廓点的值,计算最小值即可。代码如下:
class Solution {
public:
void infect(vector<vector<int>>& grid , int i , int j){
if(i<0||i==grid.size()||j<0||j==grid[0].size()||grid[i][j]!=1)return;
grid[i][j]=2;
infect(grid , i-1 , j);
infect(grid , i+1 , j);
infect(grid , i , j-1);
infect(grid , i , j+1);
}
int cal(vector<vector<int>>& grid , int i , int j){
int res = INT_MAX;
for(int ii = 0 ; ii<grid.size() ; ii++){
for(int jj = 0 ; jj<grid[0].size() ; jj++){
if(grid[ii][jj]==1){
res = min(res,abs(i-ii)+abs(j-jj)-1);
}
}
}
return res;
}
bool decide(vector<vector<int>>& grid , int i , int j){
if(i==0||i==grid.size()-1||j==0||j==grid[0].size()-1)return true;
else{
int temp = grid[i-1][j]+grid[i+1][j]+grid[i][j-1]+grid[i][j+1];
if(temp==8)return false;
}
return true;
}
int shortestBridge(vector<vector<int>>& grid){
for(int i = 0 ; i<grid.size() ; i++){
for(int j = 0 ; j<grid[0].size() ; j++){
if(grid[i][j]==1){
infect(grid,i,j);
goto here;
}
}
}
here:
int res = INT_MAX;
for(int i = 0 ; i<grid.size() ; i++){
for(int j = 0 ; j<grid[0].size() ; j++){
if(grid[i][j]==2&&decide(grid,i,j)){
int temp = cal(grid , i , j);
res = min(res , temp);
}
}
}
return res;
}
};
执行用时:388 ms, 在所有 C++ 提交中击败了5.16%的用户
内存消耗:17 MB, 在所有 C++ 提交中击败了99.35%的用户
时间复杂度非常不行,所有广度优先遍历搜索,还是首先将其中一个岛标记为2,然后不断向外扩,直到碰到1,这个时候,扩的层数为答案,代码如下:
class Solution {
public:
int shortestBridge(vector<vector<int>>& grid) {
int n = grid.size();
vector<vector<int>> dirs = {{-1, 0}, {1, 0}, {0, 1}, {0, -1}};
vector<pair<int, int>> island;
queue<pair<int, int>> qu;
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
if (grid[i][j] == 1) {
qu.emplace(i, j);
grid[i][j] = -1;
while (!qu.empty()) {
auto [x, y] = qu.front();
qu.pop();
island.emplace_back(x, y);
for (int k = 0; k < 4; k++) {
int nx = x + dirs[k][0];
int ny = y + dirs[k][1];
if (nx >= 0 && ny >= 0 && nx < n && ny < n && grid[nx][ny] == 1) {
qu.emplace(nx, ny);
grid[nx][ny] = -1;
}
}
}
for (auto &&[x, y] : island) {
qu.emplace(x, y);
}
int step = 0;
while (!qu.empty()) {
int sz = qu.size();
for (int i = 0; i < sz; i++) {
auto [x, y] = qu.front();
qu.pop();
for (int k = 0; k < 4; k++) {
int nx = x + dirs[k][0];
int ny = y + dirs[k][1];
if (nx >= 0 && ny >= 0 && nx < n && ny < n) {
if (grid[nx][ny] == 0) {
qu.emplace(nx, ny);
grid[nx][ny] = -1;
} else if (grid[nx][ny] == 1) {
return step;
}
}
}
}
step++;
}
}
}
}
return 0;
}
};
执行用时:40 ms, 在所有 C++ 提交中击败了50.98%的用户
内存消耗:19.5 MB, 在所有 C++ 提交中击败了29.63%的用户