Enumerable#Zip 实现一下

脑洞的蜂蜜

发布于 2018-02-01 19:08:54

5620

发布于 2018-02-01 19:08:54

文章被收录于专栏：NetCore

早上看到“geff Zhang”介绍了Enumerable#Zip，闲来没事弄一个实现。

也谈不上思路，看了张兄的测试代码，先写了个简单的结构：

public static IEnumerable<TResult> Zip<TFirst, TSencond, TResult>(
            this IEnumerable<TFirst> first, 
            IEnumerable<TSencond> sencond, 
            Func<TFirst, TSencond, TResult> func)
        {
            
        }

有了结构，实现就简单多了。张兄在文中介绍到“方法将第一个序列中的每个元素与第二个序列中有相同索引的元素合并。如果该序列不具有相同数目的元素，则直到它到达其中一个的末尾，该方法才合并序列。例如，如果一个序列有三个元素，另一个序列具有四个元素，那么结果序列将只有三个元素。”

要读到两个序列的各个元素，按照相同索引的元素，执行func，原先想想用foreach，for之类的，好像有点麻烦，为了简单，就直接转换成IEnumerator。

上代码，大家看了就明白了：

public static IEnumerable<TResult> Zip<TFirst, TSencond, TResult>(
            this IEnumerable<TFirst> first, 
            IEnumerable<TSencond> sencond, 
            Func<TFirst, TSencond, TResult> func)
        {
            var firsttor = first.GetEnumerator();
            var sencondtor = sencond.GetEnumerator();

            while (firsttor.MoveNext())
            {
                if (sencondtor.MoveNext())
                {
                    yield return func(firsttor.Current, sencondtor.Current);
                }
            }
        }

随便写的，也没在意性能方面，等大家一起讨论下好了。

附上测试代码：

int[] numberic = new int[] { 1, 2, 3, 4 };
            string[] words = new string[] { "a", "b", "c", "d" };
            int[] numbericDiff = new int[] { 1, 2, 3 };

            var result = numberic.Zip(words, (a, b) => a + " " + b);

            foreach (var r in result)
            {
                Console.WriteLine(r);
            }

            result = numbericDiff.Zip(words, (a, b) => a + " " + b);

            foreach (var r in result)
            {
                Console.WriteLine(r);
            }

结果：