早上看到“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); }
結果:
PS:推一個旅遊網站,大家工作之餘也要放鬆自己嘛。
http://www.sh-bus.com