IEnumerable和使用收益回报的递归


307

我有一种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>,如果此方法返回type ,则代码会生成,但是在输出中返回错误的类型。

有没有yield return同时使用递归的使用方法?


1
stackoverflow.com/questions/1815497/…:链接到线程“枚举不是固有IEnumerable的集合?”上的“ mrydengrens”答案。他的示例代码基于Eric Lippert的博客文章,向您展示了如何在Linq的递归枚举中使用堆栈,从而避免了迭代器使用昂贵的内存。恕我直言非常有用!
BillW 2010年

顺便说一句。if(c.Controls.Count > 0)-> if(c.Controls.Any()),特别是如果您
也要

我认为这种情况不会从收益中受益。为了完整起见,我提供了不带的实现yield。请看下面:)而且也是
单线的

您应小心避免yield return在递归函数中使用,否则内存使用量会急剧增加。参见stackoverflow.com/a/30300257/284795
Panic Panic

Answers:


485

在返回的方法内IEnumerable<T>yield return必须返回T,而不是返回IEnumerable<T>

更换

yield return c.GetDeepControlsByType<T>();

与:

foreach (var x in c.GetDeepControlsByType<T>())
{
  yield return x;
}

98

您需要产生递归调用产生的每个项目

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;
            }
        }
    }
}

请注意,以这种方式进行递归会产生成本-您最终将创建大量的迭代器,如果您的控制树非常深,则可能会导致性能问题。如果要避免这种情况,则基本上需要在方法中自己进行递归,以确保仅创建一个迭代器(状态机)。有关更多详细信息和示例实现,请参见此问题 -但这显然也增加了一定数量的复杂性。


2
我感到惊讶的是,在关于屈服乔恩的c.Controls.Count > 0.Any()
话题中

@Tymek实际上是在链接答案中提到的。

28

正如乔恩·斯基特(Jon Skeet)和上校Panic(Panonic 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);
}

为了使这个(优秀的)代码起作用,我不得不使用'OfType来将ControlCollection转换为IEnumerable形式。在Windows窗体中,ControlCollection不可枚举:返回control.Controls.OfType <Control>()。RecursiveSelect <Control>(c => c.Controls.OfType <Control>()).Where(c => c是T );
BillW

17

其他人为您提供了正确的答案,但我认为您的案子不会因屈服而受益。

这是一个无需屈服即可达到相同效果的代码段。

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>()));
}

2
也不使用LINQ yield吗?;)
Philipp M

圆滑的 我一直为附加foreach循环所困扰。现在,我可以使用纯函数式编程来做到这一点!
jsuddsjr 2014年

1
就可读性而言,我喜欢此解决方案,但是与使用yield相比,迭代器面临着相同的性能问题。@PhilippM:验证LINQ用途产生referencesource.microsoft.com/System.Core/R/...
赫尔曼

竖起大拇指,这是一个很好的解决方案。
Tomer W '18

12

您需要在第二秒从枚举器中返回项目,而不是枚举器本身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;
            }
        }
    }
}

9

我认为您必须让值返回枚举中的每个控件。

    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 childControl in c.GetDeepControlsByType<T>())
                {
                    yield return childControl;
                }
            }
        }
    }

8

Seredynski的语法是正确的,但是您应该小心避免yield return使用递归函数,因为这对内存使用造成了灾难。请参阅https://stackoverflow.com/a/3970171/284795,它会随着深度而爆炸性扩展(类似的功能是使用我应用程序中10%的内存)。

一种简单的解决方案是使用一个列表,并通过递归https://codereview.stackexchange.com/a/5651/754传递它

/// <summary>
/// Append the descendents of tree to the given list.
/// </summary>
private void AppendDescendents(Tree tree, List<Tree> descendents)
{
    foreach (var child in tree.Children)
    {
        descendents.Add(child);
        AppendDescendents(child, descendents);
    }
}

或者,您可以使用堆栈和while循环来消除递归调用 https://codereview.stackexchange.com/a/5661/754


0

尽管有很多好的答案,但我仍然要补充说,可以使用LINQ方法来完成同一件事。

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

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

三年前发布使用相同方法的解决方案。
Servy

@Servy尽管它很相似(我在写所有答案时都错过了所有答案……但还是这样),但还是有所不同,因为它使用.OfType <>进行过滤,并使用.Union()
yoel halb

2
OfType是不是一个真正的meainingful不同。最多只是一个小的风格上的改变。一个控件不能是多个控件的子控件,因此遍历的树已经不存在了。使用Union代替代替Concat是不必要地验证已经保证是唯一的序列的唯一性,因此是客观的降级。
Servy
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.