[LeetCode]Shortest Distance from All Buildings

对每个房子bfs,找出每个房子到所有空地的距离。用一个record二维数组来记录每个空地上所有房子到该空地的距离和。同时用另一个二维数组记录每个空地能被几个房子访问到,来保证所有房子都可以到达的了该空地。

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

 

posted @ 2015-12-14 14:02  Weizheng_Love_Coding  阅读(511)  评论(0编辑  收藏  举报