LeetCode994. 腐烂的橘子(C++中等题)

题目

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

值 0 代表空单元格；
值 1 代表新鲜橘子；
值 2 代表腐烂的橘子。
每分钟，腐烂的橘子 周围 4 个方向上相邻 的新鲜橘子都会腐烂。

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

示例 1：

输入：grid = [[2,1,1],[1,1,0],[0,1,1]]
输出：4
示例 2：

输入：grid = [[2,1,1],[0,1,1],[1,0,1]]
输出：-1
解释：左下角的橘子（第 2 行， 第 0 列）永远不会腐烂，因为腐烂只会发生在 4 个正向上。
示例 3：

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

提示：

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

思路

BFS，传播的轮次起始就是距离腐烂橘子的路径长度。腐烂橘子可能不止一个，因此我们需要先将腐烂的橘子插入到队列中，然后进行遍历。初始化时记录好橘子的个数cnt值，然后如果被腐化了，则cnt--，最终若cnt>0则返回-1。

代码

class Solution {
public:
    int orangesRotting(vector<vector<int>>& grid) {
        int d[20][20];//距离，传播的轮次
        memset(d, -1, sizeof d);
        int m = grid.size(), n = grid[0].size();
        int cnt = 0;//好橘子的个数
        int res = 0;//传播的轮次计数
        queue<pair<int, int>> q;
        //将所有烂橘子作为源点进行遍历
        for(int i = 0; i < m; i ++)
            for(int j = 0; j < n; j ++)
                {
                    if(grid[i][j] == 2)
                    {
                        d[i][j] = 0;
                        q.push({i, j});
                    }
                    else if(grid[i][j] == 1) cnt ++;//好橘子个数加一
                }

        //进行遍历
        while(q.size())
        {
            auto t = q.front();
            q.pop();
            int x = t.first, y = t.second;
            int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
            for(int i = 0; i < 4; i ++)//遍历四个方位
            {
                int a = dx[i] + x, b = dy[i] + y;
                if(a >= 0 && a < m && b >= 0 && b < n && grid[a][b] == 1 && d[a][b] == -1)
                {
                    d[a][b] = d[x][y] + 1;
                    res = max(res, d[a][b]);
                    q.push({a, b});
                    cnt --;
                }
            }
        }
        if(cnt > 0) return -1;
        return res;
    }
};