LeetCode994 腐烂的橘子

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

值 0 代表空单元格；
值 1 代表新鲜橘子；
值 2 代表腐烂的橘子。
每分钟，任何与腐烂的橘子（在 4 个正方向上）相邻的新鲜橘子都会腐烂。

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

做广度优先搜索，先遍历记录下来新鲜橘子的个数，同时把腐烂橘子坐标都压入队列。用一个数组记录每个点的腐烂时间，初始腐烂的位0，其余全为-1。不断从队列中取出腐烂节点，判断上下左右四个节点，如果符合要求就把他腐烂掉，同时更新腐烂时间。等队列为空时判断还有没有新鲜橘子就可以了。

class Solution {
private:
    int dis[10][10];
    int dir_x[4]={0, 1, 0, -1};
    int dir_y[4]={1, 0, -1, 0};

public:
    int orangesRotting(vector<vector<int>>& grid) {
        int row=grid.size();
        if(!row)
            return 0;
        int col=grid[0].size();
        int fresh=0;
        int ans=0;
        
        queue<pair<int,int>> q;
        for(int i=0;i<row;++i)
            for(int j=0;j<col;++j){
                if(grid[i][j]==1)
                    ++fresh;
                else if(grid[i][j]==2){
                    q.push(make_pair(i,j));
                }
                dis[i][j]=grid[i][j]==2?0:-1;
            }
        if(!fresh)
            return 0;
        while(!q.empty()){
            pair<int,int>pos=q.front();q.pop();
            for(int i=0;i<4;++i){
                int tx=pos.first+dir_x[i];
                int ty=pos.second+dir_y[i];
                if(tx<0 || tx>=row || ty<0 || ty>=col || !grid[tx][ty] || dis[tx][ty]!=-1)
                    continue;
                dis[tx][ty]=dis[pos.first][pos.second]+1;
                q.push(make_pair(tx,ty));
                --fresh;
                ans=dis[tx][ty];
                if(!fresh)
                    break;
            }
        }
        return fresh==0?ans:-1;
    }
};