公司备案证查询网站查询系统企业邮箱格式

leetcode:102. 沉没孤岛

题目

题目描述：

给定一个由 1（陆地）和 0（水）组成的矩阵，岛屿指的是由水平或垂直方向上相邻的陆地单元格组成的区域，且完全被水域单元格包围。孤岛是那些位于矩阵内部、所有单元格都不接触边缘的岛屿。

现在你需要将所有孤岛“沉没”，即将孤岛中的所有陆地单元格（1）转变为水域单元格（0）。

输入描述：

第一行包含两个整数 N, M，表示矩阵的行数和列数。

之后 N 行，每行包含 M 个数字，数字为 1 或者 0，表示岛屿的单元格。

输出描述

输出将孤岛“沉没”之后的岛屿矩阵。

输入示例：

4 5
1 1 0 0 0
1 1 0 0 0
0 0 1 0 0
0 0 0 1 1

输出示例：

1 1 0 0 0
1 1 0 0 0
0 0 0 0 0
0 0 0 1 1

思路

这道题跟求孤岛面积很像，我们还是从边缘的岛屿来考虑，把边缘的岛屿全部做上标记（不是0和1，这里用2来表示它是靠边的，不是中间的），然后遍历整个网格，如果遇到了1，说明就是中间的孤岛，把它变成0.

最后一步：把之前标记为2的，全部返回标记成1！

#include <bits/stdc++.h>
using namespace std;int dir[4][2] = {0, 1, 1, 0, 0, -1, -1, 0};// 将周围岛屿标记（注意是标记为2）
void dfs(vector<vector<int>> &grid, int x, int y)
{// 终止条件（这里可能是遇到海水，还可能是遇到我们之前标记过的2！！！）if (grid[x][y] == 0 || grid[x][y] == 2)return;// 处理当前节点grid[x][y] = 2;// 处理延伸节点for (int i = 0; i < 4; i++){int nextx = x + dir[i][0];int nexty = y + dir[i][1];if (nextx < 0 || nextx >= grid.size() || nexty < 0 || nexty >= grid[0].size())continue;dfs(grid, nextx, nexty);}
}int main()
{int n, m;cin >> n >> m;vector<vector<int>> grid(n, vector<int>(m, 0));for (int i = 0; i < n; i++){for (int j = 0; j < m; j++){cin >> grid[i][j];}}// 遍历四周，将非孤岛进行标记for (int i = 0; i < n; i++){if (grid[i][0] == 1)dfs(grid, i, 0);if (grid[i][m - 1] == 1)dfs(grid, i, m - 1);}for (int j = 0; j < m; j++){if (grid[0][j] == 1)dfs(grid, 0, j);if (grid[n - 1][j])dfs(grid, n - 1, j);}// 现在孤岛是1，把它变成0for (int i = 0; i < n; i++){for (int j = 0; j < m; j++){if (grid[i][j] == 1)grid[i][j] = 0;}}// 把非孤岛2全部还原成1for (int i = 0; i < n; i++){for (int j = 0; j < m; j++){if (grid[i][j] == 2)grid[i][j] = 1;}}// 打印最终的网格for (int i = 0; i < n; i++){for (int j = 0; j < m; j++){cout << grid[i][j] << " ";}cout << endl;}return 0;
}

总结

dfs终止条件那部分，注意需要加上遇到2的情况，不然就会把所有2又全部算进去了。

参考资料

 代码随想录