[Leetcode] 286. Walls and Gates

最新推荐文章于 2021-01-22 05:49:00 发布

原创最新推荐文章于 2021-01-22 05:49:00 发布 · 352 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

294 篇文章

订阅专栏

本文介绍了一种使用BFS（广度优先搜索）算法解决迷宫问题的方法，从起点开始遍历可达节点，更新最短路径。相较于DFS，BFS在此类问题上具有更优性能。

这一题其实就是一条很基础的BFS寻图题。就是由一个起点出发然后遍历周边可以走的点。DFS其实也可以做，但是BFS做起来会有更好的PERFORMANCE。就是先收集起点，然后往外扩散有效的点（也就是没走过的，值为INF的点)。

    public static int[][] DIR = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};

    public void wallsAndGates(int[][] rooms) {
        Queue<int[]> bfsQ = new LinkedList<int[]>();
        for (int i = 0; i < rooms.length; i++) {
            for (int j = 0; j < rooms[0].length; j++) {
                if (rooms[i][j] == 0) {
                    bfsQ.add(new int[]{i, j});
                }
            }
        }
        
        while (!bfsQ.isEmpty()) {
            int[] curPt = bfsQ.poll();
            int x = curPt[0], y = curPt[1];
            for (int i = 0; i < Solution.DIR.length; i++) {
                int nextX = x + Solution.DIR[i][0], nextY = y + Solution.DIR[i][1];
                if (nextX >= rooms.length || nextX < 0 || nextY >= rooms[0].length || nextY < 0 || rooms[nextX][nextY] != Integer.MAX_VALUE) {
                    continue;
                }
                
                rooms[nextX][nextY] = rooms[x][y] + 1;
                bfsQ.add(new int[]{nextX, nextY});
            }
        }
    }