2012-08-05 78 views
0

考虑下面的类,用于构建一个树层次结构的特定级别:获取所有节点在树上

public class simpletest 
{ 
    public simpletest Parent { get; set; } 
    public IList<simpletest> Children { get; set; } 
    public string Name { get; set; } 
} 

我怎么能得到一个特定级别的所有对象的列表?

下面是一些示例数据(希望其正确格式化):

var root = new simpletest() { Name = "Root" }; 
var level1childA = new simpletest() { Name = "level1childA", Parent = root }; 
var level1childB = new simpletest() { Name = "level1childB", Parent = root }; 

var level2childA = new simpletest() { Name = "level2childA", Parent = level1childA }; 
var level2childB = new simpletest() { Name = "level2childB", Parent = level1childA }; 
var level2childC = new simpletest() { Name = "level2childC", Parent = level1childB }; 
var level2childD = new simpletest() { Name = "level3childD", Parent = level1childB }; 

var level3childA = new simpletest() { Name = "level3childA", Parent = level2childA }; 
var level3childB = new simpletest() { Name = "level3childB", Parent = level2childA }; 
var level3childC = new simpletest() { Name = "level3childC", Parent = level2childB }; 
var level3childD = new simpletest() { Name = "level3childD", Parent = level2childB }; 
var level3childE = new simpletest() { Name = "level3childE", Parent = level2childC }; 
var level3childF = new simpletest() { Name = "level3childF", Parent = level2childC }; 
var level3childG = new simpletest() { Name = "level3childG", Parent = level3childD }; 
var level3childH = new simpletest() { Name = "level3childH", Parent = level3childD }; 

level2childA.Children = new List<simpletest> { level3childA, level3childB }; 
level2childB.Children = new List<simpletest> { level3childC, level3childD }; 
level2childC.Children = new List<simpletest> { level3childE, level3childF }; 
level2childD.Children = new List<simpletest> { level3childG, level3childH }; 

level1childA.Children = new List<simpletest> { level2childA, level2childB }; 
level1childB.Children = new List<simpletest> { level2childC, level2childD }; 

因此,例如,使用节点level3childE,我怎么会从同级别的所有节点?节点名称可以是任何东西。

+0

创建方法'Depth',对于每个节点返回其深度到根?然后LINQ用适当的查询选择根节点? – 2012-08-05 13:37:07

+2

你可以创建一个Level/Depth属性,然后根据这个属性查询对象,或者创建一个基本的for循环,它遍历你的树项目到期望的级别,在这一点上它选择这些对象 – timothyclifford 2012-08-05 13:39:44

回答

1

这里是作为SimpleTest的成员函数实现的(未测试的)递归方法:

IEnumerable<simpletest> ElementsAtDepth(int depth) { 
    if(depth > 0) { 
     foreach(simpletest child in this.Children) 
      foreach(simpletest element in child.ElementsAtDepth(depth - 1)) 
       yield return element; 
    } 
    else { 
     foreach(simpletest element in this.Children) 
      yield return element; 
    } 
} 
+0

感谢BlackBear,给定我的样本数据确实获得了指定深度的所有节点。我确实注意到结果包含重复项,深度为1,则返回两个level1childA和两个level1childB。此外,如果深度指定为3(最底层并且节点没有子节点(叶节点?)),则会在“foreach(this.Children中的简单测试元素)”行上生成错误。 – 2012-08-05 14:52:45

+0

@JohnSmith:我在代码中发现了一个错字,现在就试试。至于错误我相信是因为Children为null而发生的,所以你应该在costructor中初始化它。 – BlackBear 2012-08-05 15:30:42

+0

非常感谢BlackBear。 – 2012-08-05 16:25:04