C# 怎么使用 yield return 实现迭代器_C# yield return 迭代器使用教程

yield return 可实现延迟执行,如遍历 GetOddNumbers 时每次按需生成奇数,避免内存浪费。

在 C# 中,yield return 是一种简洁高效的方式来实现自定义迭代器,而无需手动创建集合或实现 IEnumerableIEnumerator 接口。它让方法在每次返回一个元素后暂停执行,下次继续从暂停处恢复,非常适合处理大数据集或需要延迟计算的场景。

yield return 基本语法

使用 yield return 的方法必须返回以下类型之一:

  • IEnumerable
  • IEnumerator
  • IEnumerable
  • IEnumerator

方法中每遇到一次 yield return,就会向枚举器提供一个值,并暂时退出。当枚举器请求下一个元素时,执行从中断处继续。

public static IEnumerable GetNumbers()
{
    yield return 1;
    yield return 2;
    yield return 3;
    yield return 4;
}

调用该方法并遍历:

foreach (int num in GetNumbers())
{
    Console.WriteLine(num);
}
// 输出:
// 1
// 2
// 3
// 4

延迟执行与内存优化

yield return 最大的优势是延迟执行(Lazy Evaluation):元素只在被请求时才生成,而不是一次性全部计算并存储在内存中。

例如,生成 1 到 n 的奇数序列:

public static IEnumerable GetOddNumbers(int max)
{
    for (int i = 1; i <= max; i += 2)
    {
        yield return i;
    }
}

即使 max 是 100 万,也不会占用大量内存,因为数字是按需生成的。

实际应用场景示例

常见用途包括遍历树结构、读取大文件、分页数据等。

比如遍历二叉树的中序序列:

public class TreeNode
{
    public int Value;
    public TreeNode Left;
    public TreeNode Right;
public IEnumerable InOrder()
{
    if (Left != null)
        foreach (int val in Left.InOrder())
            yield return val;

    yield return Value;

    if (Right != null)
        foreach (int val in Right.InOrder())
            yield return val;
}

}

这种方式代码清晰,且不会一次性构建完整的结果列表。

注意事项与限制

  • 不能在匿名方法、lambda 表达式或 try 块中含有 catch 子句的环境中使用 yield return
  • 可以配合 try-finally 使用,用于资源清理。
  • 方法一旦开始迭代,内部状态由编译器自动生成的状态机管理。
  • 如果多次枚举同一个 yield return 方法结果,每次都会重新执行逻辑。

基本上就这些。合理使用 yield return 能写出更高效、更易读的迭代代码,尤其适合处理流式数据或无限序列。掌握它,能让你的 C# 编程更加灵活。