分层类的迭代器

Question

我有一个 C# 类来定义层次结构（比下面的示例复杂得多）。该类具有同一类的Parent 和可能的Children。

我正在尝试为顶级对象编写一个迭代器，它允许我访问 foreach 循环中的所有内容。

class Node
{
    public Node Parent { get; private set; }
    public bool HasParent { get { return (Parent != null); } }
    public string Name { get; private set; }
    public bool IsAnimal { get; set; }
    public bool IsCar { get; set; }
    public List<Node> Children { get; private set; }
    public bool HasChildren { get { return (Children != null); } }

}

想这样访问：

foreach (Node myNode in TopNode.Contents)

最好有多个迭代器来遍历不同类型的Children，比如：

foreach (Node myNode in TopNode.Animals)

或

foreach (Node myNode in TopNode.Cars)

Answer 1

将此方法添加到Node 类中：

public IEnumerable<Node> DescendantsAndSelf()
{
    yield return this;
    if (Children != null) {
        foreach (Node child in Children) {
            foreach (Node node in child.DescendantsAndSelf()) {
                yield return node;
            }
        }
    }
}

而且您不需要针对不同类型的节点使用不同的迭代器。只需使用.Where(...)

var allAnimals = myTopNode.DescendantsAndSelf()
    .Where(n => n.IsAnimal);

如果您采纳@LasseVågsætherKarlsen 的建议并从抽象基类Node 派生不同的节点类型，那么您可以像这样将动物类型化为Animal：

IEnumerable<Animal> allAnimals = myTopNode.DescendantsAndSelf()
    .OfType<Animal>();

---

您也可以将Children 声明为：

public List<Node> Children { get; } = new List<Node>();

像这样，Children 永远不会为空，HasChildren 将被实现为：

public bool HasChildren => Children.Count > 0;

见：

• Iterators (C#)
• Sample LINQ Queries（辅导老师）。