c# - 递归与树中的收益返回元素顺序

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 并检索 Preorder 中的元素（在本例中为 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;
        }