3妹:2哥,今日都立冬了, 可是天气一点都不冷。
2哥 : 立冬了,晚上要不要一起出去吃饺子?🥟
3妹:好呀好呀,2哥请吃饺子喽
2哥 : 歪歪,我说的是一起出去吃,没说我请客好吧
3妹:哼,2哥真小气,请吃顿饺子都不肯!
2哥:这样,我们找一道算法题,后做出来的要请吃饺子,怎么样?
3妹:who 怕who, 来就来!
题目:给你一个下标从 0 开始大小为 m x n 的二维整数数组 grid ,它表示一个网格图。每个格子为下面 3 个值之一:
0 表示草地。
1 表示着火的格子。
2 表示一座墙,你跟火都不能通过这个格子。
一开始你在最左上角的格子 (0, 0) ,你想要到达最右下角的安全屋格子 (m - 1, n - 1) 。每一分钟,你可以移动到 相邻 的草地格子。每次你移动 之后 ,着火的格子会扩散到所有不是墙的 相邻 格子。
请你返回你在初始位置可以停留的 最多 分钟数,且停留完这段时间后你还能安全到达安全屋。如果无法实现,请你返回 -1 。如果不管你在初始位置停留多久,你 总是 能到达安全屋,请你返回 109 。
注意,如果你到达安全屋后,火马上到了安全屋,这视为你能够安全到达安全屋。
如果两个格子有共同边,那么它们为 相邻 格子。
示例 1:
image.png
输入:grid = [[0,2,0,0,0,0,0],[0,0,0,2,2,1,0],[0,2,0,0,1,2,0],[0,0,2,2,2,0,2],[0,0,0,0,0,0,0]]
输出:3
解释:上图展示了你在初始位置停留 3 分钟后的情形。
你仍然可以安全到达安全屋。
停留超过 3 分钟会让你无法安全到达安全屋。
示例 2:
image.png
输入:grid = [[0,0,0,0],[0,1,2,0],[0,2,0,0]]
输出:-1
解释:上图展示了你马上开始朝安全屋移动的情形。
火会蔓延到你可以移动的所有格子,所以无法安全到达安全屋。
所以返回 -1 。
提示:
m == grid.length
n == grid[i].length
2 <= m, n <= 300
4 <= m * n <= 2 * 10^4
grid[i][j] 是 0 ,1 或者 2 。
grid[0][0] == grid[m - 1][n - 1] == 0
思路:
思考二分查找, 具体见代码中注释:
java代码:
class Solution {
static final int INF = 1000000000;
static int[][] dirs = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
public int maximumMinutes(int[][] grid) {
int m = grid.length, n = grid[0].length;
int[][] fireTime = new int[m][n];
for (int i = 0; i < m; i++) {
Arrays.fill(fireTime[i], INF);
}
/* 通过 bfs 求出每个格子着火的时间 */
bfs(grid, fireTime);
/* 二分查找找到最大停留时间 */
int ans = -1;
int low = 0, high = m * n;
while (low <= high) {
int mid = low + (high - low) / 2;
if (check(fireTime, grid, mid)) {
ans = mid;
low = mid + 1;
} else {
high = mid - 1;
}
}
return ans >= m * n ? INF : ans;
}
public void bfs(int[][] grid, int[][] fireTime) {
int m = grid.length;
int n = grid[0].length;
Queue<int[]> queue = new ArrayDeque<int[]>();
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (grid[i][j] == 1) {
queue.offer(new int[]{i, j});
fireTime[i][j] = 0;
}
}
}
int time = 1;
while (!queue.isEmpty()) {
int sz = queue.size();
for (int i = 0; i < sz; i++) {
int[] arr = queue.poll();
int cx = arr[0], cy = arr[1];
for (int j = 0; j < 4; j++) {
int nx = cx + dirs[j][0];
int ny = cy + dirs[j][1];
if (nx >= 0 && ny >= 0 && nx < m && ny < n) {
if (grid[nx][ny] == 2 || fireTime[nx][ny] != INF) {
continue;
}
queue.offer(new int[]{nx, ny});
fireTime[nx][ny] = time;
}
}
}
time++;
}
}
public boolean check(int[][] fireTime, int[][] grid, int stayTime) {
int m = fireTime.length;
int n = fireTime[0].length;
boolean[][] visit = new boolean[m][n];
Queue<int[]> queue = new ArrayDeque<int[]>();
queue.offer(new int[]{0, 0, stayTime});
visit[0][0] = true;
while (!queue.isEmpty()) {
int[] arr = queue.poll();
int cx = arr[0], cy = arr[1], time = arr[2];
for (int i = 0; i < 4; i++) {
int nx = cx + dirs[i][0];
int ny = cy + dirs[i][1];
if (nx >= 0 && ny >= 0 && nx < m && ny < n) {
if (visit[nx][ny] || grid[nx][ny] == 2) {
continue;
}
/* 到达安全屋 */
if (nx == m - 1 && ny == n - 1) {
return fireTime[nx][ny] >= time + 1;
}
/* 火未到达当前位置 */
if (fireTime[nx][ny] > time + 1) {
queue.offer(new int[]{nx, ny, time + 1});
visit[nx][ny] = true;
}
}
}
}
return false;
}
}
网友评论