Recherche de tous les objets de type T dans une arborescence C#

Question

Je dois écrire une méthode de recherche arborescente qui prend un paramètre de type T et renvoie tous les éléments de type T qui existent dans l'arborescence. Est-ce qu'il y a un moyen de faire ça? Je préférerais l'élégance à l'efficacité à ce stade ...

Answer 1

Quelque chose comme ceci:

internal static IEnumerable<T> AllDescendantNodes<T>(this TreeNode input) 
    where T class; 
{ 
    T current = null; 
    foreach (TreeNode node in input.Nodes) 
     if((current = node as T) != null) 
     { 
      yield return current; 
      foreach (var subnode in node.AllDescendantNodes<T>()) 
       yield return subnode; 
     } 
} 

Vous pouvez ensuite appeler cela contre le nœud racine comme une méthode d'extension:

foreach(MyCustomNodeClass item in rootNode.AllDescendantNodes<MyCustomNodeClass>()) 
{ 
    ... 
} 

Answer 2

En supposant que votre arbre est générique. c'est-à-dire Item<T>.

int count = yourTree.Count(p => p == typeof(T)); 

Sinon, analyser chaque noeud et comparer « item == typeof(T) »

Answer 3

Eh bien, en interne la méthode aurait pour itérer sur tous les éléments de l'arbre, de sorte que le saut juste énumérer dessus, et en utilisant le OfType méthode LINQ est pas loin:

var onlyTs = yourTree.OfType<SomeT>(); 

Answer 4

Qu'est-ce que vous avez besoin est une fonction traversal arbre de base (pré-commande, afinde ou postorder - ce n'a pas d'importance) et une fonction de filtre. Ensuite, vous pouvez composer ces deux ensemble et obtenir ce dont vous avez besoin:

IEnumerable<T> Traverse(Tree<T> tree) 
{ 
    yield return tree.Data; 

    foreach(Tree<T> subtree in tree.Subtrees) 
     foreach(T t in Traverse(subtree)) 
      yield return t; 
} 

IEnumerable<U> Filter<T, U>(IEnumerable<T> source)   
    where U : T 
{ 
    foreach(T t in source) 
     if(t is U) 
      yield return (U)t; 
}