https://leetcode.com/problems/walls-and-gates/
You are given a m x n 2D grid initialized with these three possible values.
-1
- A wall or an obstacle.0
- A gate.INF
- Infinity means an empty room. We use the value231 - 1 = 2147483647
to representINF
as you may assume that the distance to a gate is less than2147483647
.
Fill each empty room with the distance to its nearest gate. If it is impossible to reach a gate, it should be filled with
INF
.
For example, given the 2D grid:
INF -1 0 INF INF INF INF -1 INF -1 INF -1 0 -1 INF INF
After running your function, the 2D grid should be:
3 -1 0 1 2 2 1 -1 1 -1 2 -1 0 -1 3 4
相关题: http://rainykat.blogspot.com/2017/01/leetcode-200-number-of-islandsback.html
思路: find all gates use dfs to traverse all surrounding pos, when reach wall, gate or confirm min distance, return;
Complexity: O(M*N) M: row length, N: col len (wortst case, only one gate all other rooms)
关键字: backtracking
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 i, int j,int d) { if(i<0||i>=rooms.length||j>=rooms[0].length||j<0||rooms[i][j]<d)//ensure min dis + avoid wall return;// <d: return wall + gate + ensure min distance rooms[i][j]=d;//i row,j col dfs(rooms, i - 1, j,d+1); dfs(rooms, i + 1, j,d+1); dfs(rooms, i, j - 1,d+1); dfs(rooms, i, j + 1,d+1); } }
No comments:
Post a Comment