辅助方法

Question

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

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

乔的孩子收藏只有一层深，但玛丽的孩子还有孩子，他们也有孩子。

我试图使用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);

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

Answer 1

辅助方法

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

性能问题

如果性能是任何问题，请考虑上面的LINQ /功能实现，或者看看Servy的答案，其中任何一个应该比使用yield ...的版本更有效。

Answer 2

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

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

这是一般模型，通常用于遍历树。请注意，这将进行深度优先搜索。如果您希望首先进行呼吸搜索，则可以使用Queue<T>代替Stack<T>。

如何最好地穿越儿童的孩子到一个未知的深度？

2 个答案:

辅助方法

用法

评论

性能问题