文章目錄

• 1、題目描述
• 2、解題思路
• 3、解題代碼

1、題目描述

2、解題思路

??本題需要對每一個建筑物分別進行一次 BFS。

??BFS 的作用是計算每一個空格到這個建筑物的距離。

??定義一個數組 totalDist[][]，其中 totalDist[i][j] 表示坐標 (i,j) 的空格到達所有建筑物的距離之和。

??因此，每一次 BFS，都是對 totalDist[][] 進行累加的過程。

??這里還需要注意一點的是：建筑物個數為 n，因此每一個空格都會被遍歷 n，次，因為要計算空格到達每一個建筑物的距離嘛。

??因此，我們需要定義一個變量 mark = 0，初始時，空格是用 mark 表示，當我計算完一個建筑物時，mark–，所有的空格變成用 -1 表示了。

??這樣做就是為了解決“BFS時空格不能重復遍歷，但是遍歷建筑物時，空格又要重復使用”的矛盾。

3、解題代碼

class Solution {private final int[][] directions = {{0, -1}, {0, 1}, {-1, 0}, {1, 0}};// 思路：// (1) 從每一個建筑物開始進行廣度優先搜索// (2) 在搜索的同時計算每一個空格到這個建筑物的距離// (3) 在搜索的同時將每一個空格到每一個建筑物的距離進行累加，得到每個空格到所有建筑物的距離// (4) 取空格到所有建筑物的最小距離public int shortestDistance(int[][] grid) {int rows = grid.length;if (rows == 0) {return 0;}int cols = grid[0].length;int[][] totalDist = new int[rows][cols];int ans = Integer.MAX_VALUE;// 用于標記空地int mark = 0;for (int i = 0; i < rows; i++) {for (int j = 0; j < cols; j++) {// (1) 從每一個建筑物開始進行廣度優先搜索if (grid[i][j] == 1) {ans = bfs(grid, rows, cols, i, j, mark, totalDist);// 每次遍歷搜索完一個建筑物，這個標記減一，表示所有空地被遍歷一次了mark--;}}}return ans;}private int bfs(int[][] grid, int rows, int cols, int x, int y, int mark, int[][] totalDist) {int ans = Integer.MAX_VALUE;Queue<int[]> queue = new LinkedList<>();// 隊列中每個數組有 3 個元素，分別表示：// 第一個元素和第二個元素表示坐標值// 第三個元素表示當前坐標到建筑物的距離// 第三個元素的初始值為 0 的原因是：一開始的時候從當前建筑物到當前建筑物的距離是 0queue.add(new int[]{x, y, 0});while (!queue.isEmpty()) {int[] curPoint = queue.poll();int currDist = curPoint[2];for (int[] direction : directions) {int newX = curPoint[0] + direction[0];int newY = curPoint[1] + direction[1];if (newX >= 0 && newX < rows && newY >= 0 && newY < cols && // 不超出邊界grid[newX][newY] == mark) { // 當前點是空地// (2) 在搜索的同時計算每一個空格到這個建筑物的距離int dist = currDist + 1;// (3) 在搜索的同時將每一個空格到每一個建筑物的距離進行累加totalDist[newX][newY] += dist;// (4) 取空格到所有建筑物的最小距離ans = Math.min(ans, totalDist[newX][newY]);queue.add(new int[]{newX, newY, dist});// 當前點處理過了grid[newX][newY]--;}}}return ans == Integer.MAX_VALUE ? -1 : ans;} }

總結

以上是生活随笔為你收集整理的【LeetCode - 317】离建筑物最近的距离的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。