力扣994——腐烂的橘子（多源BFS）

最新推荐文章于 2024-09-24 08:39:28 发布

lllzzzhhh2589

最新推荐文章于 2024-09-24 08:39:28 发布

阅读量180

点赞数

分类专栏：算法文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/lllzzzhhh2589/article/details/123282663

版权

算法专栏收录该内容

185 篇文章 0 订阅

订阅专栏

这是一个关于算法实现的问题，代码描述了一种解决方案，利用广度优先搜索（BFS）来模拟腐烂橘子在网格中扩散的过程。从所有腐烂的橘子位置开始，每一步更新相邻的未腐烂橘子状态，直到所有橘子都腐烂或无法继续传播。函数返回腐烂过程所需的最短时间，如果不能全部腐烂则返回-1。

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

题目链接

多源BFS，从每个腐烂的橘子同时出发，类似层序遍历，层数就是感染时间

class Solution {
public:
    vector<pair<int, int>> dir = {{-1,0}, {1,0}, {0,-1}, {0,1}};
    int orangesRotting(vector<vector<int>>& grid) {
        int m = grid.size(), n = grid[0].size(), ans = -1, flag = 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) q.push({i, j});
                if(grid[i][j]) flag = 1;
            }
        }
        if(flag == 0) return 0;
        while(!q.empty()) {
            int tmpn = q.size();
            ans++;
            while(tmpn--) {
                auto [x, y] = q.front(); q.pop();
                for(auto [dx, dy] : dir) {
                    int nx = x + dx, ny = y + dy;
                    if(nx < 0 || nx >= m || ny < 0 || ny >= n) continue;
                    if(grid[nx][ny] == 1) {
                        grid[nx][ny] = 2;
                        q.push({nx, ny});
                    }
                }
            }
        }
        for(int i = 0; i < m; i++) {
            for(int j = 0; j < n; j++) {
                if(grid[i][j] == 1) return -1;
            }
        }
        return ans;
    }
};