Question

我目前有一棵树，我正在使用Functionally traversing a tree in C#中的代码来获取树中的路径。问题是，此代码当前返回树中每个路径的列表。例如，如果我有以下树：

A---
    |
    ---B
    |
    ---C
       |
       ---D
       |
       ---E

代码将返回{A}，{A，B}，{A，C}，{A，C，D}，{A，C，E}

我实际上需要它才能返回分支，即{A，B}，{A，C，D}，{A，C，E}。

这并不是上述方法的一部分，只要最终我得到的列表只是分支而不是每条路径。

我真正想弄清楚的是如何从上面的方法过滤IEnumerable列表，只有每个列表中最后一个元素没有子元素的条目。

所以在我目前的代码中：

class Node
{
     public string Name;
     public int ParentID;
     public List<Node> Children;
}

然后在我的主要代码中

List<Node> listOfNodes = getTreeNodes();

Node rootNode = listOfNodes.Where(n => n.ParentID == 0).FirstOrDefault();
// the below paths var will have every path instead of just the branches
var paths = ComputePaths(rootNode, n=>n.Children);

使用链接答案中的代码：

static IEnumerable<IEnumerable<T>> ComputePaths<T>(T Root, Func<T, IEnumerable<T>> Children) {
    yield return new[] { Root };
    foreach (var Child in Children(Root)) 
        foreach (var ChildPath in ComputePaths(Child, Children)) 
            yield return new[] { Root }.Concat(ChildPath);            
}

希望这是有道理的。任何帮助将不胜感激。

Answer 1

这有效：

static IEnumerable<IEnumerable<T>> ComputePaths<T>(T Root, Func<T, IEnumerable<T>> Children) {
    var children = Children(Root);
    if (children != null && children.Any())
    {
        foreach (var Child in children) 
            foreach (var ChildPath in ComputePaths(Child, Children)) 
                yield return new[] { Root }.Concat(ChildPath);            
    } else {
        yield return new[] { Root };
    }
}

如果节点是叶节点（没有子节点），我们返回自己。否则，我们会返回其子女的路径。

检索树路径列表

1 个答案: