给你一个 m * n 的网格,其中每个单元格不是 0(空)就是 1(障碍物)。每一步,您都可以在空白单元格中上、下、左、右移动。
如果您 最多 可以消除 k 个障碍物,请找出从左上角 (0, 0) 到右下角 (m-1, n-1) 的最短路径,并返回通过该路径所需的步数。如果找不到这样的路径,则返回 -1。
示例 1:
输入:
grid =
[[0,0,0],
[1,1,0],
[0,0,0],
[0,1,1],
[0,0,0]],
k = 1
输出:6
解释:
不消除任何障碍的最短路径是 10。
消除位置 (3,2) 处的障碍后,最短路径是 6 。该路径是 (0,0) -> (0,1) -> (0,2) -> (1,2) -> (2,2) -> (3,2) -> (4,2).
示例 2:
输入:
grid =
[[0,1,1],
[1,1,1],
[1,0,0]],
k = 1
输出:-1
解释:
我们至少需要消除两个障碍才能找到这样的路径。
提示:
grid.length == m
grid[0].length == n
1 <= m, n <= 40
1 <= k <= m*n
grid[i][j] == 0 or 1
grid[0][0] == grid[m-1][n-1] == 0
思路:广搜模板题!
class Solution {
class node{
int x,y,value,step;
public node(int x,int y,int step,int value) {
this.x=x;
this.y=y;
this.step=step;
this.value=value;
}
}
int[] dx= {1,-1,0,0};
int[] dy= {0,0,1,-1};
public int shortestPath(int[][] grid, int k) {
int m=grid.length;
int n=grid[0].length;
boolean[][][] flag=new boolean[m][n][k+1];
Queue<node> q=new LinkedList<>();
q.add(new node(0,0,0,0));
flag[0][0][0]=true;
while(!q.isEmpty()) {
node now=q.poll();
//System.out.println(now.x+" "+now.y);
if(now.x==m-1 && now.y==n-1)
return now.step;
for(int i=0;i<4;i++) {
int x=now.x+dx[i];
int y=now.y+dy[i];
//System.out.println(x+" "+y);
if(x<0 || x>=m || y<0 || y>=n)
continue;
int value=now.value+(grid[x][y]==1?1:0);
if(value>k || flag[x][y][value])
continue;
flag[x][y][value]=true;
q.add(new node(x,y,now.step+1,value));
}
}
return -1;
}
}