Leetcode—994. 腐烂的橘子【中等】

源代码•宸

已于 2024-05-14 09:18:08 修改

阅读量234

点赞数 6

分类专栏： LeetCode刷题文章标签： leetcode 算法职场和发展经验分享 c++

于 2024-05-14 09:17:23 首次发布

本文链接：https://blog.csdn.net/qq_44631615/article/details/138835982

版权

LeetCode刷题专栏收录该内容

330 篇文章 5 订阅

订阅专栏

2024每日刷题（136）

Leetcode—994. 腐烂的橘子

在这里插入图片描述

实现代码

class Solution {
public:
    int orangesRotting(vector<vector<int>>& grid) {
        queue<pair<int, int>> q;
        int ans = 0;
        constexpr int dirs[4][2] = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}};
        int m = grid.size();
        int n = grid[0].size();
        int countFresh = 0;

        for(int i = 0; i < m; i++) {
            for(int j = 0; j < n; j++) {
                if(grid[i][j] == 1) {
                    countFresh++;
                } else if(grid[i][j] == 2) {
                    q.emplace(i, j);
                }
            }
        }

        if(countFresh == 0) {
            return 0;
        }

        while(!q.empty()) {
            ans++;
            for(int sz = q.size(); sz > 0; sz--) {
                auto [i, j] = q.front();
                q.pop();
                for(auto &[dx, dy]: dirs) {
                    int x = i + dx;
                    int y = j + dy;
                    if(x < 0 || x == m || y == n || y < 0) {
                        continue;
                    }
                    if(grid[x][y] != 1) {
                        continue;
                    }
                    grid[x][y] = 2;
                    countFresh--;
                    q.emplace(x, y);
                }
            }
        }
        return ans = (countFresh > 0) ? -1 : ans - 1;
    }
};