图论

最新推荐文章于 2024-08-16 18:21:24 发布

TheJayChou

最新推荐文章于 2024-08-16 18:21:24 发布

阅读量120

点赞数 1

分类专栏：算法和数据结构 Java

本文链接：https://blog.csdn.net/qq_32843395/article/details/113957124

版权

Java 同时被 2 个专栏收录

12 篇文章 0 订阅

订阅专栏

算法和数据结构

7 篇文章 0 订阅

订阅专栏

该博客介绍了一种使用多源广度优先搜索(BFS)算法来计算二维矩阵中每个格子到最近水域距离的方法。在给定的矩阵中，高度表示到水的距离，从每个水域开始进行搜索，利用BFS遍历并更新每个点的距离。通过一个邻接方向数组进行移动，并在搜索过程中更新距离并加入队列。最终返回一个二维数组，其中每个元素表示对应位置到最近水源的距离。

摘要由CSDN通过智能技术生成

快速幂
多源bfs
在这里插入图片描述

// 多源bfs 每个格子的高度等于它到离它最近的水的距离
class Solution {
    public int[][] highestPeak(int[][] isWater) {
        int m = isWater.length, n = isWater[0].length;
        Queue<int[]> q = new LinkedList();
        int[][] d = new int[m][n];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (isWater[i][j] == 1) {
                    d[i][j] = 0;
                    q.offer(new int[] {i, j});	// 从每个水域（一个源）开始搜索
                } else d[i][j] = -1;
            }
        }
        int[] dx = {-1, 0, 1, 0}, dy = {0, 1, 0, -1};	// 上，右，下，左
        // 第i轮搜索到的点，距最近水源的距离即为i（水源为第0轮），也为上一轮搜到的距离+1
        while (!q.isEmpty()) {
            int[] t = q.poll();
            for (int i = 0; i < 4; i++) {
                int x = t[0] + dx[i], y = t[1] + dy[i];
                if (x < 0 || x >= m || y < 0 || y >= n) continue;
                if (d[x][y] == -1) {
                    d[x][y] = d[t[0]][t[1]] + 1;
                    q.offer(new int[] {x, y});
                }
            }
        }
        return d;
    }
}