IEnumerable和使用yield返回的递归

我有一个IEnumerable<T>方法，我正在使用它来查找WebForms页面中的控件。

这个方法是递归的，当yield return返回递归调用的值时，我在返回我想要的类型时遇到了一些问题。

我的代码如下:

public static IEnumerable<Control> GetDeepControlsByType<T>(this Control control)
{
    foreach(Control c in control.Controls)
    {
        if (c is T)
        {
            yield return c;
        }

        if(c.Controls.Count > 0)
        {
            yield return c.GetDeepControlsByType<T>();
        }
    }
}

当前抛出“无法转换表达式类型”错误。但是，如果此方法返回类型IEnumerable<Object>，则构建代码，但在输出中返回错误的类型。

有没有一种方法可以在使用递归的同时使用收益率?

当前回答

别人给了你正确的答案，但我不认为让步对你的案子有好处。

这里有一个片段，它达到了同样的目的而不让步。

public static IEnumerable<Control> GetDeepControlsByType<T>(this Control control)
{
   return control.Controls
                 .Where(c => c is T)
                 .Concat(control.Controls
                                .SelectMany(c =>c.GetDeepControlsByType<T>()));
}

2013-08-11 14:38:29

其他回答

虽然有很多好的答案，但我仍然想补充一点，使用LINQ方法来完成同样的事情是可能的。

例如，OP的原始代码可以重写为:

public static IEnumerable<Control> 
                           GetDeepControlsByType<T>(this Control control)
{
   return control.Controls.OfType<T>()
          .Union(control.Controls.SelectMany(c => c.GetDeepControlsByType<T>()));        
}

2016-06-21 15:07:20

在第二个yield return中，需要从枚举器返回项，而不是枚举器本身

public static IEnumerable<Control> GetDeepControlsByType<T>(this Control control)
{
    foreach (Control c in control.Controls)
    {
        if (c is T)
        {
            yield return c;
        }

        if (c.Controls.Count > 0)
        {
            foreach (Control ctrl in c.GetDeepControlsByType<T>())
            {
                yield return ctrl;
            }
        }
    }
}

2010-01-13 10:30:39

你需要生成由递归调用产生的每一项:

public static IEnumerable<Control> GetDeepControlsByType<T>(this Control control)
{
    foreach(Control c in control.Controls)
    {
        if (c is T)
        {
            yield return c;
        }

        if(c.Controls.Count > 0)
        {
            foreach (Control control in c.GetDeepControlsByType<T>())
            {
                yield return control;
            }
        }
    }
}

注意，以这种方式递归是有代价的——你最终会创建很多迭代器，如果你有一个非常深的控制树，这可能会产生性能问题。如果希望避免这种情况，基本上需要自己在方法中进行递归，以确保只创建了一个迭代器(状态机)。有关更多细节和示例实现，请参阅这个问题-但这显然也增加了一定程度的复杂性。

2010-01-13 10:29:57

别人给了你正确的答案，但我不认为让步对你的案子有好处。

这里有一个片段，它达到了同样的目的而不让步。

public static IEnumerable<Control> GetDeepControlsByType<T>(this Control control)
{
   return control.Controls
                 .Where(c => c is T)
                 .Concat(control.Controls
                                .SelectMany(c =>c.GetDeepControlsByType<T>()));
}

2013-08-11 14:38:29

正如Jon Skeet和Colonel Panic在他们的回答中指出的，如果树很深，在递归方法中使用yield return可能会导致性能问题。

下面是一个通用的非递归扩展方法，它执行深度优先遍历树序列:

public static IEnumerable<TSource> RecursiveSelect<TSource>(
    this IEnumerable<TSource> source, Func<TSource, IEnumerable<TSource>> childSelector)
{
    var stack = new Stack<IEnumerator<TSource>>();
    var enumerator = source.GetEnumerator();

    try
    {
        while (true)
        {
            if (enumerator.MoveNext())
            {
                TSource element = enumerator.Current;
                yield return element;

                stack.Push(enumerator);
                enumerator = childSelector(element).GetEnumerator();
            }
            else if (stack.Count > 0)
            {
                enumerator.Dispose();
                enumerator = stack.Pop();
            }
            else
            {
                yield break;
            }
        }
    }
    finally
    {
        enumerator.Dispose();

        while (stack.Count > 0) // Clean up in case of an exception.
        {
            enumerator = stack.Pop();
            enumerator.Dispose();
        }
    }
}

与Eric Lippert的解决方案不同，RecursiveSelect直接使用枚举器，因此不需要调用Reverse(这将在内存中缓冲整个序列)。

使用RecursiveSelect, OP的原始方法可以像这样简单地重写:

public static IEnumerable<Control> GetDeepControlsByType<T>(this Control control)
{
    return control.Controls.RecursiveSelect(c => c.Controls).Where(c => c is T);
}

2015-05-25 15:21:57

IEnumerable和使用yield返回的递归

推荐文章

最新文章

标签