[LeetCode]Shortest Distance from All Buildings

對每一個房子bfs,找出每一個房子到全部空地的距離。用一個record二維數組來記錄每一個空地上全部房子到該空地的距離和。同時用另外一個二維數組記錄每一個空地能被幾個房子訪問到,來保證全部房子均可以到達的了該空地。node

public class Solution {
    public int shortestDistance(int[][] grid) {
        int row = grid.length;
        if (row == 0) {
            return -1;
        }
        int col = grid[0].length;
        int[][] record1 = new int[row][col]; // visited num
        int[][] record2 = new int[row][col]; // distance
        int num1 = 0;
        for (int r = 0; r < row; r++) {
            for (int c = 0; c < col; c++) {
                if (grid[r][c] == 1) {
                    num1 ++;
                    boolean[][] visited = new boolean[row][col];
                    Queue<int[]> queue = new LinkedList<int[]>();
                    queue.offer(new int[]{r, c});
                    int dist = 0;
                    while (!queue.isEmpty()) {
                        int size = queue.size();
                        for (int i = 0; i < size; i++) {
                            int[] node = queue.poll();
                            int x = node[0];
                            int y = node[1];
                            record2[x][y] += dist;
                            record1[x][y] ++;
                            if (x > 0 && grid[x - 1][y] == 0 && !visited[x - 1][y]) {
                                queue.offer(new int[]{x - 1, y});
                                visited[x - 1][y] = true;
                            }
                            if (x + 1 < row && grid[x + 1][y] == 0 && !visited[x + 1][y]) {
                                queue.offer(new int[]{x + 1, y});
                                visited[x + 1][y] = true;
                            }
                            if (y > 0 && grid[x][y - 1] == 0 && !visited[x][y - 1]) {
                                queue.offer(new int[]{x, y - 1});
                                visited[x][y - 1] = true;
                            }
                            if (y + 1 < col && grid[x][y + 1] == 0 && !visited[x][y + 1]) {
                                queue.offer(new int[]{x, y + 1});
                                visited[x][y + 1] = true;
                            }
                        }
                        dist ++;
                    }
                }
            }
        }
        int result = Integer.MAX_VALUE;
        for (int r = 0; r < row; r++) {
            for (int c = 0; c < col; c++) {
                if (grid[r][c] == 0 && record1[r][c] == num1 && record2[r][c] < result) {
                    result = record2[r][c];
                }
            }
        }
        return result == Integer.MAX_VALUE ? -1 : result;
    }
}
相關文章
相關標籤/搜索