其他分享
首页 > 其他分享> > leetcode 腐烂的橘子

leetcode 腐烂的橘子

作者:互联网

994. 腐烂的橘子

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

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

提示:

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

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.这样我们可以省略最后一个循环查找是否还存在新鲜橘子。

标签:int,腐烂,ij,que,nex,grid,橘子,leetcode
来源: https://blog.csdn.net/qq_22684941/article/details/120501694