
我讨厌图论。
这道题写了特别久,不过好歹也是写出来了......
方法是先将橘子全部遍历一遍,做两件事:①找出所有连通的橘子②找出所有腐烂的橘子,设置一个vector<queue<int>>,每个vector元素代表一片连通的橘子,其中的queue代表这片连通橘子中腐烂的橘子,这样当即就可以判断有没有橘子永远不会腐烂。
这里的int要表示行和列,我用了上一题并查集给我的思路,就是用row*列数+colume只用int就可以记录当前行列。
然后再遍历我们设置的vector<queue<int>>,每次取出队列头元素,遍历四个方向,将四个方向可以被腐烂的橘子全部加入队列,如果没有可以遍历的橘子了就比较最终时间和结果result,然后遍历下一片连通橘子。
代码挺长的:
cpp
class Solution {
public:
int result=0;
vector<queue<int>> rot;
vector<vector<bool>> pass;
vector<vector<int>> grid;
void dfs(int i,int j){
if(i>=0&&j>=0&&i<grid.size()&&j<grid[0].size()&&grid[i][j]!=0&&pass[i][j]==0) pass[i][j]=1;
else return;
if(grid[i][j]==2) rot[rot.size()-1].push(i*grid[0].size()+j);
dfs(i+1,j);
dfs(i-1,j);
dfs(i,j+1);
dfs(i,j-1);
}
bool spread(int i,int j,int row){
if(i>=0&&j>=0&&i<grid.size()&&j<grid[0].size()&&grid[i][j]==1){
grid[i][j]=2;
rot[row].push(i*grid[0].size()+j);
return 1;
}
else return 0;
}
int orangesRotting(vector<vector<int>>& grid) {
this->grid=grid;
this->pass=vector<vector<bool>> (grid.size(),vector<bool> (grid[0].size(),0));
for(int i=0;i<grid.size();i++){
for(int j=0;j<grid[0].size();j++){
if(grid[i][j]!=0&&pass[i][j]==0){
rot.push_back(queue<int> {});
dfs(i,j);
if(rot[rot.size()-1].empty()) return -1;
}
}
}
int result=0;
for(int i=0;i<rot.size();i++){
int r=0;
int size=rot[i].size();
bool b=0;
while(!rot[i].empty()){
int row=rot[i].front()/grid[0].size();
int line=rot[i].front()%grid[0].size();
rot[i].pop();
size--;
b|=spread(row+1,line,i);
b|=spread(row-1,line,i);
b|=spread(row,line+1,i);
b|=spread(row,line-1,i);
if(size==0){
size=rot[i].size();
if(b==0) break;
else b=0;
r++;
}
}
result=max(result,r);
}
return result;
}
};