Двоичное дерево Traversal - PREORDER

Question

Двоичное дерево Traversal - PREORDER

голоса

8

Я пытаюсь реализовать обходе дерева PREORDER с использованием возвращения урожая, который возвращает IEnumerable

private IEnumerable<T> Preorder(Node<T> node)
{

    while(node != null)
    {
        yield return node.Data;
        yield return node.LeftChild.Data;
        yield return node.RightChild.Data;
    }

}

В этом случае он переходит в бесконечный цикл и да, я знаю, что мне нужно, чтобы сохранить перемещения. Как это может быть сделано?

Если LeftChild или RightChild имеет нулевое значение, бросает пустое исключение. Я думаю, что в тот момент мне нужно дать перерыв;

Я полагаю, Симметричные и postorder бы тоже похоже, какие-то идеи?

У меня есть версия Resursive, что хорошо работает.

public void PreOrderTraversal(Node<T> node)
{

    if(node!=null)
    {
        Console.Write(node.Data);
    }
    if (node.LeftChild != null)
    {
        PreOrderTraversal(node.LeftChild);
    }

    if (node.RightChild != null)
    {
        PreOrderTraversal(node.RightChild);
    }
}

Благодарю.

c#.net algorithm tree binary-search-tree

Задан 04/06/2011 в 03:34
источник пользователем DarthVader

На других языках...

1 ответов

Alex Aza · Accepted Answer · 2011-06-04 04:04

Вариант № 1 Рекурсивных

public class Node<T> : IEnumerable<T>
{
    public Node<T> LeftChild { get; set; }

    public Node<T> RightChild { get; set; }

    public T Data { get; set; }

    public IEnumerator<T> GetEnumerator()
    {
        yield return Data;

        if (LeftChild != null)
        {
            foreach (var child in LeftChild)
                yield return child;
        }
        if (RightChild != null)
        {
            foreach (var child in RightChild)
                yield return child;
        }
    }

    System.Collections.IEnumerator System.Collections.IEnumerable.GetEnumerator()
    {
        return GetEnumerator();
    }
}

Применение:

var child1 = new Node<int> { Data = 1 };
var child2 = new Node<int> { Data = 2 };
var child3 = new Node<int> { Data = 3, LeftChild = child1 };
var root = new Node<int> { Data = 4, LeftChild = child3, RightChild = child2 };

foreach (var value in root)
    Console.WriteLine(value);

Вариант № 2 нерекурсивна статический метод

public static IEnumerable<T> Preorder<T>(Node<T> root)
{
    var stack = new Stack<Node<T>>();
    stack.Push(root);

    while (stack.Count > 0)
    {
        var node = stack.Pop();
        yield return node.Data;
        if (node.RightChild != null)
            stack.Push(node.RightChild);
        if (node.LeftChild != null)
            stack.Push(node.LeftChild);
    }
}