快捷導(dǎo)航

C#?9使用foreach擴(kuò)展的示例詳解

更新時(shí)間：2023年01月06日 10:03:30 作者：Newbe36524

在?C#?9?中，foreach?循環(huán)可以使用擴(kuò)展方法。在本文中，我們將通過例子回顧?C#?9?中如何擴(kuò)展?foreach?循環(huán)，感興趣的小伙伴可以了解一下

代碼演示

下面是一個(gè)對樹形結(jié)構(gòu)進(jìn)行深度優(yōu)先遍歷的示例代碼：

using System;
using System.Collections.Generic;
 
namespace Example
{
    class TreeNode
    {
        public int Value { get; set; }
        public List<TreeNode> Children { get; set; }
 
        public TreeNode(int value)
        {
            Value = value;
            Children = new List<TreeNode>();
        }
    }
 
    static class TreeExtensions
    {
        public static IEnumerable<TreeNode> DepthFirst(this TreeNode root)
        {
            yield return root;
            foreach (var child in root.Children.SelectMany(DepthFirst))
            {
                yield return child;
            }
        }
    }
 
    class Program
    {
        static void Main(string[] args)
        {
            var root = new TreeNode(1);
            root.Children.Add(new TreeNode(2));
            root.Children.Add(new TreeNode(3));
            root.Children[0].Children.Add(new TreeNode(4));
            root.Children[0].Children.Add(new TreeNode(5));
 
            foreach (var node in root.DepthFirst())
            {
                Console.WriteLine(node.Value);
            }
            // Outputs: 1 2 4 5 3
        }
    }
}

在這個(gè)示例代碼中，我們在 TreeNode 類中定義了一個(gè)值屬性和一個(gè)存儲子節(jié)點(diǎn)的列表屬性。我們還在 TreeExtensions 類中定義了一個(gè) DepthFirst 擴(kuò)展方法，該方法使用 yield return 語句來返回樹形結(jié)構(gòu)的深度優(yōu)先遍歷結(jié)果。

在 Main 方法中，我們創(chuàng)建了一個(gè)樹形結(jié)構(gòu)，然后使用 foreach 循環(huán)來遍歷樹形結(jié)構(gòu)的深度優(yōu)先遍歷結(jié)果。

之所以使用擴(kuò)展方法往往是因?yàn)椋覀兛梢栽诓恍薷?TreeNode 類的情況下，為 TreeNode 類添加新的功能。

那么接下來我們希望在 C# 9 中默認(rèn)為 TreeNode 類添加 DepthFirst 行為，這樣我們就可以直接使用 foreach 循環(huán)來遍歷樹形結(jié)構(gòu)的深度優(yōu)先遍歷結(jié)果了。

C# 9 中的 foreach 擴(kuò)展

在 C# 9 中，我們可以使用 foreach 擴(kuò)展來實(shí)現(xiàn)上面的需求。我們只需要在 TreeNode 類中添加一個(gè) GetEnumerator 方法，該方法返回一個(gè)實(shí)現(xiàn)了 IEnumerable 接口的對象即可。

static class TreeExtensions
{
    public static IEnumerable<TreeNode> DepthFirst(this TreeNode root)
    {
        yield return root;
        foreach (var child in root.Children.SelectMany(DepthFirst))
        {
            yield return child;
        }
    }
 
    public static IEnumerator<TreeNode> GetEnumerator(this TreeNode root)
    {
        return root.DepthFirst().GetEnumerator();
    }
}

在上面的代碼中，我們在 TreeNode 類中添加了一個(gè) GetEnumerator 方法，該方法返回一個(gè)實(shí)現(xiàn)了 IEnumerable 接口的對象。這個(gè)對象就是我們在 DepthFirst 方法中使用 yield return 語句返回的結(jié)果。

現(xiàn)在我們可以直接使用 foreach 循環(huán)來遍歷樹形結(jié)構(gòu)的深度優(yōu)先遍歷結(jié)果了。

foreach (var node in root)
{
    Console.WriteLine(node.Value);
}

總結(jié)

在 C# 9 中，我們可以使用 foreach 擴(kuò)展來為類添加新的行為。在上面的示例代碼中，我們?yōu)?TreeNode 類添加了 DepthFirst 行為，這樣我們就可以直接使用 foreach 循環(huán)來遍歷樹形結(jié)構(gòu)的深度優(yōu)先遍歷結(jié)果了。

到此這篇關(guān)于C# 9使用foreach擴(kuò)展的示例詳解的文章就介紹到這了,更多相關(guān)C# 9 foreach擴(kuò)展內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: