3

想象一个具有以下属性的对象：

        class TestObject
        {
            public string Name { get; set; }
            public Collection<TestObject> Children { get; set; }
        }

现在以锯齿状方式初始化一些：

var person1 = new TestObject(){ 
                          Name = "Joe", 
                          Children = new Collection<TestObject>(){ childCollection1 }; 
                              };

var person2 = new TestObject(){ 
                          Name = "Mary", 
                          Children = new Collection<TestObject>(){ childCollection2 }; 
                              };

其中 Joe 的 childCollection 只深了一层，但 Mary 的孩子有孩子，他们也有孩子。

我试图使用 SelectMany 没有运气。

// Works
var joe = person1.Children.SelectMany(c => c.Children).Concat(person1.Children);

// Does not work - only returns 1 level deep
var mary = person2.Children.SelectMany(c => c.Children).Concat(person2.Children);

将包含每个孩子的结果检索到未知深度的最佳方法是什么？

4

2 回答 2

5

辅助方法

public static IEnumerable<T> Traversal<T>(
    T root,
    Func<T, IEnumerable<T>> getChildren)
{
    if (root == null)
    {
        yield break;
    }

    yield return root;

    var children = getChildren(root);
    if (children == null)
    {
        yield break;
    }

    foreach (var child in children)
    {
        foreach (var node in Traversal(child, getChildren))
        {
            yield return node;
        }
    }
}

//Or if you don't need all those null checks, here's a more compact version.
public static IEnumerable<T> Traversal<T>(
    T root,
    Func<T, IEnumerable<T>> getChildren)
{
    yield return root;
    foreach (var child in getChildren(root))
        foreach (var node in Traversal(child, getChildren))
            yield return node;
}

//If you like a LINQ/functional style better, this is also equivalent.
public static IEnumerable<T> Traversal<T>(
    T root,
    Func<T, IEnumerable<T>> getChildren)
{
    return new T[] { root }
        .Concat(getChildren(root)
            .SelectMany(child => Traversal(child, getChildren)));
}

用法

var everybody = Traversal(person, x => x.Children);

注释

您可以轻松地修改该Traversal方法以完全按照您想要的方式运行。例如，如果您只想要叶节点，那么您应该只yield return root;在children为空或为空时。

性能问题

如果性能是任何类型的问题，请考虑上面的 LINQ/功能实现或查看 Servy 的答案，其中任何一个都应该比使用yield ....

于 2013-08-15T20:03:50.043 回答
5

您可以像这样编写通用遍历方法：

public static IEnumerable<T> Traverse<T>(T root, 
    Func<T, IEnumerable<T>> childSelector)
{
    var stack = new Stack<T>();
    stack.Push(root);

    while (stack.Any())
    {
        var next = stack.Pop();
        yield return next;
        foreach (var child in childSelector(next))
            stack.Push(child);
    }
}

这是一个通用模型，通常对遍历树很有用。请注意，这将进行深度优先搜索。如果您想先进行呼吸搜索，则可以使用 aQueue<T>而不是 a Stack<T>。

于 2013-08-15T20:12:28.687 回答