使用linq检查树的长度

本文关键字：linq 检查 使用 | 更新日期: 2023-09-27 17:59:43

嗨，假设我有以下类型的树

public class Element
{
    public List<Element> element;
}

假设树根是

Element root = GetTree();

我知道使用递归检查这个树的长度是可能的但是使用linq可以检查这个树的长度吗？

使用linq检查树的长度

您可以编写一个扩展方法来递归地检索所有元素。

var allElements = root.element.Traverse(el => el.element);

例如：

public static IEnumerable<T> Traverse<T>(this IEnumerable<T> source, Func<T, IEnumerable<T>> fnRecurse)
{
    foreach (T item in source)
    {
        yield return item;
        IEnumerable<T> seqRecurse = fnRecurse(item);
        if (seqRecurse != null)
        {
            foreach (T itemRecurse in Traverse(seqRecurse, fnRecurse))
            {
                yield return itemRecurse;
            }
        }
    }
}

添加一个新的扩展方法；

    public static int CountX(this Element e)
    {
        int temp = 0;
        if (e.element != null)
        {
            temp = e.element.Count;
            e.element.ForEach(q => temp += q.CountX());
        }
        return temp;
    }

并称之为like；

int depthCount= a.CountX();

据我所知，您不可能拥有可回避的Linq，因为拥有递归lambda是不可能的。

我能给出的最原始的答案是基于递归lambda表达式，该表达式基于可重用的Fixpoint运算符。你会发现林克的大部分机制。但我担心固定点部分是没有纯粹林克答案的原因。

public static class FixPoint
{
    // Reusable fixpoint operator
    public static Func<T, TResult> Fix<T, TResult>(Func<Func<T, TResult>, Func<T, TResult>> f)
    {
        return t => f(Fix<T, TResult>(f))(t);
    }
}
public class Element
{
    public List<Element> element;

    public int CalculateMaxDepth()
    {
        return FixPoint.Fix<List<Element>, int>(
            // recursive lambda
            f =>
            listElement => listElement == null || listElement.Count == 0 
                ? 0 
                : 1 + listElement.Select(e => f(e.element)).Max())
            (this.element);
    }
    [Test]
    public  void myTest()
    {
        var elt = new Element() { element = new List<Element> { new Element() { element = new List<Element> { new Element() { element = new List<Element> { new Element() } } } } } };
        Assert.AreEqual(3, elt.CalculateMaxDepth());
    }
}
相关文章：