这也是一道典型的dfs的题。。。和number of island很像,但是又有一些特点。。。
刚刚了number of island,dfs的设计其实是为了销毁整个岛屿,起的是毁灭作用。。。主程序的话,还是在原来的方法里面实现,通过扫一遍grid,遇到了1就调用dfs
这里呢,我可以很快想到,在主程序里面,扫一遍grid,遇到了0,就调用dfs,进行铺路!!对,这里的dfs就是铺路的作用。。。那么怎么铺呢?就是设计dfs的问题了。。。
思路的关键是设置一个distance作为参数!!所以一用有 grid,x,y,distance这4个参数。。。如果原有的数比现在的distance要小那就停止,否则,设置为现在这个distance,因为这个distance是可以一定保证从0铺路铺到这的。。。基本思路就有了。。。
public class Solution {
public void wallsAndGates(int[][] rooms) {
for(int i=0; i<rooms.length; i++){
for(int j=0; j<rooms[0].length; j++){
if(rooms[i][j]==0) dfs(rooms, i, j, 0); // only start at the gate!!! road can only build from the gate!!
}
}
}
public void dfs(int[][] rooms, int x, int y, int dis){
if(x<0 || y<0 || x>=rooms.length || y>=rooms[0].length || rooms[x][y]<dis) return;
rooms[x][y]=dis;
dfs(rooms, x+1, y, dis+1); // right
dfs(rooms, x-1, y, dis+1); // left
dfs(rooms, x, y+1, dis+1); // up
dfs(rooms, x, y-1, dis+1); // down
}
}
public class Solution {
public void wallsAndGates(int[][] rooms) {
for(int i=0; i<rooms.length; i++){
for(int j=0; j<rooms[0].length; j++){
if(rooms[i][j]==0) dfs(rooms, i, j, 0);
}
}
}
private void dfs(int[][] rooms, int x, int y, int distance){
if(x<0 || y<0 || x>=rooms.length || y>= rooms[0].length || rooms[x][y]<distance) return; // 注意了不能写<=,=的时候要继续的,想想其实的时候是0啊,给出的distance 也是0
rooms[x][y]=distance; // 居然忘记了铺路这一最重要最核心的一步!!!
dfs(rooms, x-1, y, distance+1);
dfs(rooms, x+1, y, distance+1);
dfs(rooms, x, y-1, distance+1);
dfs(rooms, x, y+1, distance+1);
}
}