leetcode 腐烂的橘子

994. 腐烂的橘子

在给定的网格中,每个单元格可以有以下三个值之一:

  • 值 0 代表空单元格;
  • 值 1 代表新鲜橘子;
  • 值 2 代表腐烂的橘子。
    每分钟,任何与腐烂的橘子(在 4 个正方向上)相邻的新鲜橘子都会腐烂。

返回直到单元格中没有新鲜橘子为止所必须经过的最小分钟数。如果不可能,返回 -1。

示例 1:

输入:[[2,1,1],[1,1,0],[0,1,1]]
输出:4

示例 2:

输入:[[2,1,1],[0,1,1],[1,0,1]]
输出:-1
解释:左下角的橘子(第 2 行, 第 0 列)永远不会腐烂,因为腐烂只会发生在 4 个正向上。

示例 3:

输入:[[0,2]]
输出:0
解释:因为 0 分钟时已经没有新鲜橘子了,所以答案就是 0 。

提示:

  • 1 <= grid.length <= 10
  • 1 <= grid[0].length <= 10
  • grid[i][j] 仅为 0、1 或 2

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/rotting-oranges。

  • 广度优先搜索
    先找到所有的腐烂源,并将腐烂源和空格子标记,避免重复访问。然后使用广度优先搜索以腐烂源为起点依次往外搜索,每次时间加1.
class Solution {
public:
    int orangesRotting(vector<vector<int>>& grid) {
        std::queue<pair<int, int>> que_ij;
        vector<vector<int>> find_ij(grid.size(), vector<int>(grid[0].size(), 0));
        for (int i = 0; i < grid.size(); ++i) {
            for (int j = 0; j < grid[0].size(); ++j) {
                if (grid[i][j] == 2) {
                    que_ij.emplace(i,j);
                    find_ij[i][j] = 1;
                }
                if (grid[i][j] == 0) {
                    find_ij[i][j] = 1;
                }
            }
        }
        std::queue<pair<int, int>> que_tmp;
        int times = 0;
        while (!que_ij.empty()) {
            auto [x, y] = que_ij.front();
            que_ij.pop();
            for (int i = 0; i < 4; ++i) {
                int nex_x = x + dx[i];
                int nex_y = y + dy[i];
                if (nex_x >= 0 && nex_x < grid.size() && nex_y >= 0 && nex_y < grid[0].size() && !find_ij[nex_x][nex_y]) {
                    find_ij[nex_x][nex_y] = 1;
                    que_tmp.emplace(nex_x, nex_y);
                }
            }
            if (que_ij.empty() && !que_tmp.empty()) {
                que_ij.swap(que_tmp);
                ++times;
            }
        }
        for (int i = 0; i < find_ij.size(); ++i) {
            for (int j = 0; j < find_ij[0].size(); ++j) {
                if (find_ij[i][j] == 0) {
                    times = -1;
                }
            }
        }
        return times;
    }
private:
    const int dx[4] = {0, 0, -1, 1};
    const int dy[4] = {1, -1, 0, 0};
};

优化:我们可以在第一次遍历时记录下新鲜橘子的数量,每次被腐蚀一个后-1.这样我们可以省略最后一个循环查找是否还存在新鲜橘子。

上一篇:剑指 Offer 21. 调整数组顺序使奇数位于偶数前面


下一篇:HDU - 1495 非常可乐(搜索)