美文网首页
【教3妹学编程-算法题】逃离火灾

【教3妹学编程-算法题】逃离火灾

作者: 程序员小2 | 来源:发表于2023-11-08 08:00 被阅读0次
    今日立冬

    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;
        }
    }
    

    相关文章

      网友评论

          本文标题:【教3妹学编程-算法题】逃离火灾

          本文链接:https://www.haomeiwen.com/subject/vmqqwdtx.html