【力扣】994.腐烂的橘子

时间:2024-02-29 17:39:06

 思路:这个题目一看就是用BFS法,并结合队列的使用,但是我不知道怎么用。嗯,后面会用了。具体就是先遍历一遍二维数组,得出所有新鲜橘子的数量,然后将烂掉的橘子收入队列中,后面再往四个方向进行腐烂的操作。再将已经腐烂掉的橘子收入队列中即可。

1、临界条件:直到没有新鲜橘子

2、用一个二维数组来控制其四周的移动

        vector<pair<int, int>> dirs = { {-1, 0}, {1, 0}, {0, -1}, {0, 1} };

代码如下:

class Solution {
public:
    int orangesRotting(vector<vector<int>>& grid) {
        int min = 0,fresh = 0;
        queue<pair<int, int>> q;
        // 统计新鲜橘子的数量并将腐烂的橘子放入队列中
        for(int i = 0;i < grid.size();i++){
            for(int j = 0;j < grid[0].size();j++){
                if(grid[i][j] == 1) fresh++;
                else if(grid[i][j] == 2) q.push({i,j});
            }
        }
        //开始腐烂
        vector<pair<int, int>> dirs = { {-1, 0}, {1, 0}, {0, -1}, {0, 1} };
        while(!q.empty()){
            int n = q.size();
            bool rotten = false; //用以判断是否发生了腐烂的这个行为
            // 这个操作是把每一次腐烂的橘子收进来,同时吐出上一次腐烂的橘子
            for(int i = 0; i < n; i++){
                auto x = q.front();
                q.pop();
                for(auto cur: dirs){
                    int i = x.first + cur.first;
                    int j = x.second + cur.second;
                    if(i >= 0 && i < grid.size() && j>=0 && j < grid[0].size() && grid[i][j] == 1){
                        grid[i][j] = 2;
                        q.push({i,j});
                        fresh--;
                        rotten = true;
                    }
                }
            }
            if(rotten) min++;
        }
        return fresh ? -1 : min;
    }
};