Question

我有以下算法用于广度优先搜索：

q := []
q.append(root node of tree)
while q:
    n := q.pop(0)
    yield n
        if n has children:
            c := children of node
            for i in c:
                q.append(i)

1）如何延长它以便跟踪当前深度？ 2）此扩展是否适用于深度优先搜索的类似算法，队列q被堆栈替换？

Answer 1

只需存储节点的深度，并在每次生成节点的子节点时将其递增。

q := [(root, 0)]
while q:
    n, depth := q.pop()
    yield n, depth
    if n has children:
        c := children of n
        for i in c:
            q.append(i, depth + 1)

这个想法延伸到DFS和启发式引导搜索。

Answer 2

为了扩展larsmans的优秀答案，下面是我的c++深度限制广度优先二叉树遍历的代码。

（该代码假定Node不包含深度信息，并在将每个节点排入队列之前将其包装在NodeAndDepth结构中。）

struct NodeAndDepth {
    NodeAndDepth(Node *n, unsigned int d) : node(n), depth(d) {}
    Node *node;
    unsigned int depth;
};

void traverseBreadthFirst_WithDepthLimit(Node *root, unsigned int maxDepth) {
    if (maxDepth == 0 || root == NULL) { return; }

    std::queue<NodeAndDepth> q;
    q.push(NodeAndDepth(root, 1));

    while (!q.empty()) {
        NodeAndDepth n = q.front(); q.pop();

        // visit(n.node); 
        // cout << n.depth << ": " << n.node->payload << endl;

        if (n.depth >= maxDepth) { continue; }

        if (n.node->left != NULL) {
            q.push(NodeAndDepth(n.node->left, n.depth + 1));
        }

        if (n.node->right != NULL) {
            q.push(NodeAndDepth(n.node->right, n.depth + 1));
        }
    }
}

在非递归广度优先搜索中跟踪深度

2 个答案: