IEnumerable and Recursion using yield return
我有一个
该方法是递归的,当
我的代码如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | 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>(); } } } |
当前引发"无法转换表达式类型"错误。但是,如果此方法返回类型
在使用递归的同时,有没有使用
在返回
替换
1 | yield return c.GetDeepControlsByType<T>(); |
用:
1 2 3 4 | foreach (var x in c.GetDeepControlsByType<T>()) { yield return x; } |
您需要生成递归调用生成的每个项:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | 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; } } } } |
请注意,以这种方式递归是有代价的——您最终会创建很多迭代器,如果您有一个非常深的控制树,那么这会导致性能问题。如果您想避免这种情况,您基本上需要自己在方法中进行递归,以确保只创建了一个迭代器(状态机)。请参阅这个问题了解更多细节和示例实现——但这显然也增加了一定的复杂性。
正如jon skeet和colonel panic在他们的答案中指出的,如果树非常深,在递归方法中使用
下面是一个通用的非递归扩展方法,它执行一系列树的深度优先遍历:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | 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(); } } } |
与EricLippert的解决方案不同,RecursiveSelect直接与枚举器一起工作,这样它就不需要调用Reverse(这会缓冲内存中的整个序列)。
使用recursiveselect,可以这样简单地重写op的原始方法:
1 2 3 4 | public static IEnumerable<Control> GetDeepControlsByType<T>(this Control control) { return control.Controls.RecursiveSelect(c => c.Controls).Where(c => c is T); } |
其他人给了你正确的答案,但我认为你的案子不会从屈服中获益。
这里有一个片段,它可以在不屈服的情况下实现相同的结果。
1 2 3 4 5 6 7 | 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>())); } |
您需要从第二个
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | 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; } } } } |
我认为您必须返回枚举表中的每个控件。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | 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; } } } } |
seredynski的语法是正确的,但是在递归函数中应该小心避免使用
一个简单的解决方案是使用一个列表并通过递归传递它https://codereview.stackexchange.com/a/5651/754
1 2 3 4 5 6 7 8 9 10 11 | /// <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
虽然有很多好的答案,但我还是要补充的是,使用LINQ方法来完成相同的事情是可能的。
例如,操作的原始代码可以重写为:
1 2 3 4 5 6 | public static IEnumerable<Control> GetDeepControlsByType<T>(this Control control) { return control.Controls.OfType<T>() .Union(control.Controls.SelectMany(c => c.GetDeepControlsByType<T>())); } |