Рекурсия с порядком возврата элементов в дереве

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

        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;
        }