今日不肯埋头，明日何以抬头！每日一句努力自己的话哈哈~哈喽，今天我将给大家带来一篇《改进广度优先搜索 (BFS) 的实现方法》，主要内容是讲解等等，感兴趣的朋友可以收藏或者有更好的建议在评论提出，我都会认真看的！大家一起进步，一起学习！

背景

我在 3d 空间中有 3d 体素。它们由 x, y, z 索引。它们被标记为 full 或 empty。我尝试有效地计算由邻居 full 体素组成的组件数量。

bfs 详细信息

我有以下代码来实现广度优先搜索（bfs）算法。每个体素由 [3]int{x, y, z} 表示。

// Count separate components consisting of disconnected finite elements.
func (vg *VoxelGrid) CountComponents() int {
    // Map key is (x, y, z) index of voxel.
    visited := make(map[[3]int]bool)
    count := 0
    for z := 0; z < vg.Len.Z; z++ {
        for y := 0; y < vg.Len.Y; y++ {
            for x := 0; x < vg.Len.X; x++ {
                if !visited[[3]int{x, y, z}] {
                    count++
                    vg.bfs(visited, [3]int{x, y, z})
                }
            }
        }
    }
    return count
}

// Algorithm: breadth-first search (BFS).
// This is much faster than depth first search (DFS) algorithm.
func (vg *VoxelGrid) bfs(visited map[[3]int]bool, start [3]int) {
    queue := [][3]int{start}
    visited[start] = true

    for len(queue) > 0 {
        v := queue[0]
        queue = queue[1:]

        neighbors := vg.getNeighbors(v)

        for _, n := range neighbors {
            if !visited[n] {
                visited[n] = true
                queue = append(queue, n)
            }
        }
    }
}

// It returns a list of neighbor voxels that are full, i.e. not empty.
func (vg *VoxelGrid) getNeighbors(v [3]int) [][3]int {
    var neighbors [][3]int
    for i := -1; i <= 1; i++ {
        for j := -1; j <= 1; j++ {
            for k := -1; k <= 1; k++ {
                if i == 0 && j == 0 && k == 0 {
                    continue
                }

                x := v[0] + i
                y := v[1] + j
                z := v[2] + k

                if x >= 0 && x < vg.Len.X &&
                    y >= 0 && y < vg.Len.Y &&
                    z >= 0 && z < vg.Len.Z {
                    // Index is valid.
                } else {
                    continue
                }

                // Is neighbor voxel empty or not?
                if vg.IsNotEmpty(x, y, z) {
                    neighbors = append(neighbors, [3]int{x, y, z})
                }
            }
        }
    }
    return neighbors
}

问题

上面的实现无法正常工作。对于仅应包含 8 组件的简单模型，它返回组件计数为 1224：

问题

我通过 vs code 调试器单步调试了代​​码。但我无法找出这个错误。有人在代码中看到任何可疑的东西吗？有什么提示可以指引我正确的方向吗？

正确答案

问题是您还在 empty 体素上调用 bfs。

在 countcomponents 中，已验证 bfs 仅在尚未访问的体素上调用（良好）：

if !visited[[3]int{x, y, z}] {

...但它缺少测试该体素是否是 full （不好），并且 bfs 会很乐意将其添加到队列中，期望它是 full。因此，每个 empty 体素也被计为一个（1 体素大小）组件。

终于介绍完啦！小伙伴们，这篇关于《改进广度优先搜索 (BFS) 的实现方法》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！