递归与yield返回树中的元素顺序

Question

我有一个递归函数，在给定起始根节点的情况下返回所有子树节点。

private IEnumerable<Node> getAllNodesRecursively(Node subnode)
{
    foreach (Node node in subnode.Nodes)
        getAllNodesRecursively(node);

    yield return subnode;
}

对于以下树结构：

A
|
+--B
|
+--C
|  |
|  +--D
|
+--E

当我尝试这样迭代时：

foreach (Node n in getAllNodesRecursively(a))
{
    Console.WriteLine(n);
}

该函数仅返回 A 值。

我希望将yield-return与递归结合使用并检索预购中的元素（本例中为A、B、C、D、E）。

（如果我将收益返回放在 foreach 之前，则 foreach 将永远不会发生）。

这可能吗？

Answer 1

您是否尝试过类似的操作：

private IEnumerable<Node> getAllNodesRecursively(Node subnode) 
{ 
    // Return the parent before its children
    yield return subnode; 

    foreach (Node node in subnode.Nodes) 
    {
        foreach(Node n in getAllNodesRecursively(node))
        {
            yield return n;
        }
    }
} 

您的实现正在递归调用 getAllNodesRecursively，但忽略其返回值。

Answer 2

是的，这是可能的，只需将 yield return 放在 foreach 之前即可。您正在考虑正常 return 语句的行为。

Answer 3

您需要显式迭代+yield返回每个节点ala的子节点：

        public IEnumerable<int> preOrder(Node root)
        {
            if (root == null)
                yield break;

            yield return root.val;

            if (root.left != null)
                foreach (int i in preOrder(root.left))
                    yield return i;

            if (root.right != null)
                foreach (int i in preOrder(root.right))
                    yield return i;
        }