题目来源
题目描述
class Solution {
public:
int countSubIslands(vector<vector<int>>& grid1, vector<vector<int>>& grid2) {
}
};
题目解析
DFS
这道题的关键在于:如何快速判断子岛屿。
一个方法是借助UnionFind。
现在我们用DFS来做
什么时候grid2中的一个岛屿B是grid1中的一个岛屿的子岛呢?
- 当岛屿B中所有陆地在岛屿A中也是陆地的时候,岛屿B是岛屿A的子岛屿
- 反过来说,如果岛屿B中存在一片陆地,在岛屿A的对应位置是海水,那么岛屿B就不是岛屿A的子岛。
那么,我们只要遍历grid2中的所有岛屿,把那些不可能是子岛的岛屿排除掉,剩下的就是子岛。
class Solution {
// 从 (i, j) 开始,将与之相邻的陆地都变成海水
void dfs(vector<vector<int>>& grid, int i, int j){
int m = grid.size(), n = grid[0].max_size();
if (i < 0 || j < 0 || i >= m || j >= n) {
return;
}
if (grid[i][j] == 0) {
return;
}
grid[i][j] = 0;
dfs(grid, i + 1, j);
dfs(grid, i, j + 1);
dfs(grid, i - 1, j);
dfs(grid, i, j - 1);
}
public:
int countSubIslands(vector<vector<int>>& grid1, vector<vector<int>>& grid2) {
int m = grid1.size(), n = grid1[0].max_size();
for (int i = 0; i < m; ++i) {
for (int j = 0; j < n; ++j) {
if(grid1[i][j] == 0 && grid2[i][j] == 1){
// 这个岛屿肯定不是子岛,淹掉
dfs(grid2, i, j);
}
}
}
// 现在 grid2 中剩下的岛屿都是子岛,计算岛屿数量
int res = 0;
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (grid2[i][j] == 1) {
res++;
dfs(grid2, i, j);
}
}
}
return res;
}
};