LeetCode 994. 腐烂的橘子

最新推荐文章于 2024-07-25 11:15:54 发布

海绵宝宝卖海绵

最新推荐文章于 2024-07-25 11:15:54 发布

阅读量206

点赞数 1

文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/qq_44190767/article/details/136712843

版权

文章介绍了如何使用广度优先搜索算法解决LeetCode上的腐烂的橘子问题，通过队列操作更新网格中橘子的状态并计算未腐烂的橘子数量。

摘要由CSDN通过智能技术生成

994. 腐烂的橘子https://leetcode.cn/problems/rotting-oranges/description/?envType=study-plan-v2&envId=top-100-liked广度优先遍历就完事儿~，最后的判断还有没有好橘子还能优化下，但问题不大。

class Solution {
public:
    int orangesRotting(vector<vector<int>>& grid) {
        queue<pair<int, int>> rotten_oranges;
        for (int i=0; i < grid.size(); ++i) {
            for (int j=0; j < grid[0].size(); ++j) {
                if (grid[i][j] == 2) rotten_oranges.push({i, j});
            }
        }

        int res = 0;
        queue<pair<int, int>> temp;
        while (!rotten_oranges.empty()) {
            while (!rotten_oranges.empty()) {
                int i = rotten_oranges.front().first;
                int j = rotten_oranges.front().second;
                rotten_oranges.pop();
                if (i +1<grid.size() && grid[i+1][j] == 1) {
                    temp.push({i+1, j});
                    grid[i+1][j] = 2;
                }
                if (i -1 >= 0 && grid[i-1][j] == 1) {
                    temp.push({i-1, j});
                    grid[i-1][j] = 2;
                }
                if (j + 1< grid[0].size() && grid[i][j+1] == 1) {
                    temp.push({i, j+1});
                    grid[i][j+1] = 2;
                }
                if (j - 1 >=0 && grid[i][j-1] == 1) {
                    temp.push({i, j-1});
                    grid[i][j-1] = 2;
                }
            }
            if (temp.empty()) break;

            while (!temp.empty()) {
                int i = temp.front().first;
                int j = temp.front().second;
                temp.pop();
                rotten_oranges.push({i, j});
            }
            res ++;
        }
        for (vector<int> rows: grid) {
            for (int i: rows) {
                if (i == 1) return -1;
            }
        }
        return res;

    }
};