leetcode 994. 腐烂的橘子(Rotting Oranges)

Posted 2021-02-15 zhanzq1

目录

• 题目描述：
• 示例 1：
• 示例 2：
• 示例 3：
• 解法：

题目描述：

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

• 值 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

---

解法：

# define PR pair<int, int>

class Solution {
public:
    bool valid(int m, int n, int i, int j){
        if(i < 0 || j < 0 || i >= m || j >= n){
            return false;
        }else{
            return true;
        }
    }
    int orangesRotting(vector<vector<int>>& grid) {
        int m = grid.size();
        int n = grid[0].size();
        vector<vector<bool>> visited(m, vector<bool>(n, false));
        vector<PR> lst;
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(grid[i][j] == 0){
                    visited[i][j] = true;
                }else if(grid[i][j] == 2){
                    visited[i][j] = true;
                    lst.push_back({i, j});
                }
            }
        }
        
        vector<PR> nxt;
        int step = 0;
        while(!lst.empty()){
            nxt.clear();
            for(PR pos : lst){
                int x = pos.first, y = pos.second;
                for(int dx = -1; dx <= 1; dx++){
                    for(int dy = -1; dy <= 1; dy++){
                        if(dx + dy == 1 || dx + dy == -1){
                            int _x = dx + x;
                            int _y = dy + y;
                            if(valid(m, n, _x, _y) && visited[_x][_y] == false){
                                nxt.push_back({_x, _y});
                                visited[_x][_y] = true;
                            }
                        }
                    }
                }
            }
            if(!nxt.empty()){
                step++;
            }
            lst = nxt;
        }
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(visited[i][j] == false){
                    return -1;
                }
            }
        }
        return step;
    }
};