目录
菜鸟做题,语言是 C++
解题思路:
思路说明图:
如步骤 1 所示,我们找到 “1”(红框内部),它可以作为一个岛屿的开头。接下来,我们寻找与这个 “1” 直接或间接连接在一起的 “1”,如步骤 2 所示。这一坨 “1”(红框内部)构成一个岛屿。
直接连接 是指上下左右四个方向,斜对角方向的不算。
除此之外,为了避免我们下一次寻找 “1” 时,把这座岛屿内部的 “1” 视为下一个岛屿的开头,我们要将这些 “1” 置为 “0” 。
我们是对整个二维数组进行遍历的,若不在遍历完一座岛屿后将 “1” 置为 “0”,那么这座岛屿除开头之外的 “1” 会被误认为是下一座岛屿的开头。
具体代码:
① Find “1”:在二维数组中寻找 “1”,作为岛屿的开头。
- for (int i = 0; i < nr; ++i) {
- for (int j = 0; j < nc; ++j) {
- if (grid[i][j] == '1') {
- ++count;
- helper(grid, i, j);
- }
- }
- }
nr 是二维数组的行数,nc 是二维数组的列数。一旦找到 “1” 就 ++count,即认为找到了一座新的岛屿。同时,使用 helper 函数去寻找与当前 “1” 直接或间接连接在一起的 “1” 。
② Find Island:寻找与当前 “1” 直接或间接连接在一起的 “1”,它们构成一座岛屿。
- void helper(vector
char >>& grid, int r, int c) { - int nr = grid.size();
- int nc = grid[0].size();
-
- grid[r][c] = '0';
- if (r - 1 >= 0 && grid[r - 1][c] == '1') helper(grid, r - 1, c);
- if (r + 1 < nr && grid[r + 1][c] == '1') helper(grid, r + 1, c);
- if (c - 1 >= 0 && grid[r][c - 1] == '1') helper(grid, r, c - 1);
- if (c + 1 < nc && grid[r][c + 1] == '1') helper(grid, r, c + 1);
- }
这四个 if 其实就是做上下左右四个方向的边界判断,同时判断当前 “1” 的邻居是不是 “1” 。若找到相邻的 “1”,那么再递归寻找与相邻的 “1” 直接或间接连接在一起的 “1” 。
- class Solution {
- public:
- void helper(vector
char >>& grid, int r, int c) { - int nr = grid.size();
- int nc = grid[0].size();
-
- grid[r][c] = '0';
- if (r - 1 >= 0 && grid[r - 1][c] == '1') helper(grid, r - 1, c);
- if (r + 1 < nr && grid[r + 1][c] == '1') helper(grid, r + 1, c);
- if (c - 1 >= 0 && grid[r][c - 1] == '1') helper(grid, r, c - 1);
- if (c + 1 < nc && grid[r][c + 1] == '1') helper(grid, r, c + 1);
- }
-
- int numIslands(vector
char >>& grid) { - int nr = grid.size();
- if (nr == 0) return 0;
- int nc = grid[0].size();
-
- int count = 0;
- for (int i = 0; i < nr; ++i) {
- for (int j = 0; j < nc; ++j) {
- if (grid[i][j] == '1') {
- ++count;
- helper(grid, i, j);
- }
- }
- }
- return count;
- }
- };
与 200. 岛屿数量 像又不像,区别在于是否有时间观念
解题思路:
具体代码:
① 寻找腐烂的橘子(2),与 200 题的代码几乎一样。
- for (int i = 0; i < nr; ++i) {
- for (int j = 0; j < nc; ++j) {
- if (temp[i][j] == 2)
- helper(grid, i, j);
- }
- }
② 对位于腐烂的橘子(2)四周的新鲜橘子(1)进行污染。
- void helper(vector
int >>& grid, int r, int c) { - if (r - 1 >= 0 && grid[r - 1][c] == 1) grid[r - 1][c] = 2;
- if (r + 1 < nr && grid[r + 1][c] == 1) grid[r + 1][c] = 2;
- if (c - 1 >= 0 && grid[r][c - 1] == 1) grid[r][c - 1] = 2;
- if (c + 1 < nc && grid[r][c + 1] == 1) grid[r][c + 1] = 2;
- }
③ 判断是否所有的新鲜橘子(1)都被污染。
- bool isRotted(vector
int >>& grid) { - for (int i = 0; i < nr; ++i) {
- for (int j = 0; j < nc; ++j) {
- if (grid[i][j] == 1) return false;
- }
- }
- return true;
- }
④ 判断是否无法继续污染:在进行新一轮污染之前,先把上一轮的污染结果 grid 存入 temp 中,如果这一轮污染后有 temp == grid,则说明已经无法继续污染了。
- vector
int>> temp = grid; - for (int i = 0; i < nr; ++i) {
- for (int j = 0; j < nc; ++j) {
- if (temp[i][j] == 2)
- helper(grid, i, j);
- }
- }
- if (temp == grid) return -1;
这样做还有一个好处,就是可以通过 if (temp[i][j] == 2) 来寻找腐烂的橘子,避免在这一轮中新腐烂的橘子参与到污染中。
- class Solution {
- public:
- int nr, nc;
- void helper(vector
int >>& grid, int r, int c) { - if (r - 1 >= 0 && grid[r - 1][c] == 1) grid[r - 1][c] = 2;
- if (r + 1 < nr && grid[r + 1][c] == 1) grid[r + 1][c] = 2;
- if (c - 1 >= 0 && grid[r][c - 1] == 1) grid[r][c - 1] = 2;
- if (c + 1 < nc && grid[r][c + 1] == 1) grid[r][c + 1] = 2;
- }
-
- bool isRotted(vector
int >>& grid) { - for (int i = 0; i < nr; ++i) {
- for (int j = 0; j < nc; ++j) {
- if (grid[i][j] == 1) return false;
- }
- }
- return true;
- }
-
- int orangesRotting(vector
int >>& grid) { - nr = grid.size();
- nc = grid[0].size();
-
- int count = 0;
- while (!isRotted(grid)) {
- vector
int>> temp = grid; - for (int i = 0; i < nr; ++i) {
- for (int j = 0; j < nc; ++j) {
- if (temp[i][j] == 2)
- helper(grid, i, j);
- }
- }
- if (temp == grid) return -1;
- ++count;
- if (isRotted(grid)) return count;
- }
- return count;
- }
- };
参考官方题解进行了升级,仿二叉树的层序遍历,不用像 2.1 那样每次都进行全部遍历
核心思想:将属于同一时刻的腐烂橘子视为属于同一层。
上图画出了橘子逐步腐烂的 5 个时刻,每个时刻中打红叉的腐烂橘子属于同一层,打灰叉的腐烂橘子属于上一层。
解题思路:
思路说明图:
对于时刻 1,让腐烂的橘子入队;对于时刻 2,队列中的腐烂橘子出队,让它们对四周的新鲜橘子进行污染,最后将新被污染的橘子入队。以此类推。
在一轮污染中,如果有橘子被污染,则计时器 +1,同时判断新鲜橘子是否被污染完毕;如果没有橘子被污染,则跳出循环,同时判断新鲜橘子是否被污染完毕。若没有橘子被污染且新鲜橘子没有被污染完毕,则表明无法污染所有新鲜橘子。
具体代码:
① 初始化:
- int freshCount = 0;
- queue
int, int>> q; - for (int i = 0; i < nr; ++i) {
- for (int j = 0; j < nc; ++j) {
- if (grid[i][j] == 1) {
- ++freshCount;
- } else if (grid[i][j] == 2) {
- q.push(make_pair(i, j));
- }
- }
- }
nr 是 grid 的行数,nc 是 grid 的列数。
② 循环结构和二叉树的层序遍历一模一样:
- while (!q.empty()) {
- int currentSize = q.size();
- for (int i = 0; i < currentSize; ++i) {
- pair<int, int> pos = q.front();
- q.pop();
-
- // 对橘子进行污染
- }
- }
③ 针对每个腐烂橘子,对其四周进行污染:
- for (int i = 0; i < 4; ++i) {
- int x = pos.first + dir_x[i];
- int y = pos.second + dir_y[i];
- if (x < 0|| x >= nr || y < 0|| y >= nc || grid[x][y] == 0)
- continue;
- if (grid[x][y] == 1) {
- hasPolluted = true;
- --freshCount;
- grid[x][y] = 2;
- q.push(make_pair(x, y));
- }
- if (freshCount == 0) break;
- }
hasPolluted 用于表明当前层中是否至少有一个腐烂橘子造成了污染,如果没有造成污染,那么就要考虑是否无法污染所有新鲜橘子了。
- class Solution {
- public:
- int dir_x[4] = {0, 1, 0, -1};
- int dir_y[4] = {1, 0, -1, 0};
- int orangesRotting(vector
int >>& grid) { - int nr = grid.size();
- if (nr == 0) return 0;
- int nc = grid[0].size();
-
- int freshCount = 0;
- queue
int, int>> q; - for (int i = 0; i < nr; ++i) {
- for (int j = 0; j < nc; ++j) {
- if (grid[i][j] == 1) {
- ++freshCount;
- } else if (grid[i][j] == 2) {
- q.push(make_pair(i, j));
- }
- }
- }
-
- int timeCount = 0;
- int hasPolluted = false;
- while (!q.empty()) {
- int currentSize = q.size();
- for (int i = 0; i < currentSize; ++i) {
- pair<int, int> pos = q.front();
- q.pop();
- for (int i = 0; i < 4; ++i) {
- int x = pos.first + dir_x[i];
- int y = pos.second + dir_y[i];
- if (x < 0|| x >= nr || y < 0|| y >= nc || grid[x][y] == 0)
- continue;
- if (grid[x][y] == 1) {
- hasPolluted = true;
- --freshCount;
- grid[x][y] = 2;
- q.push(make_pair(x, y));
- }
- if (freshCount == 0) break;
- }
- }
- if (hasPolluted) ++timeCount;
- hasPolluted = false;
- }
- return freshCount == 0 ? timeCount : -1;
- }
- };
技能点:使用循环结构来测试上/下/左/右四个方位。
- int dir_x[4] = {0, 1, 0, -1};
- int dir_y[4] = {1, 0, -1, 0};
-
- for (int i = 0; i < 4; ++i) {
- int x = pos.first + dir_x[i];
- int y = pos.second + dir_y[i];
- if (x < 0|| x >= nr || y < 0|| y >= nc)
- // ...
- }
并且用逆否命题来作为判断条件,就不需要写很多 && 了!