帮助实现一个ZipWithRatio扩展方法

Question

我试图找到一个优雅的方式来写一个可能被称为MergeWithRatio或ZipWithRatio的扩展方法。

我希望它有这个签名..

public static IEnumerable<T> MergeWithRatio<T>(this IEnumerable<T> source, IEnumerable<T> mergeSequence, int ratio) 
{ 
    .. 
} 

实例第一。

var source = new[]{1,2,3,4,5,6,7,8,9}; 
var mergeSeq = new[]{100,200}; 
var result = source.MergeWithRatio(mergeSeq, 3).ToArray(); 

结果现在包含[1，2，100，3，4，200，5，6，100中，7，8，200 9]

总结： 的MERGESEQ环绕时/如果完成，并且当“源”为空时，收益停止。

如果可能的话，我会优先考虑解决方案是懒惰的，但它不是一个要求。

任何想法或指针？

Answer 1

这会工作，并返回预期的输出：

Answer 2

public static IEnumerable<T> MergeWithRatio<T>(this IEnumerable<T> source, IEnumerable<T> mergeSequence, int ratio) 
{ 
    using (IEnumerator<T> sourceEnumerator = source.GetEnumerator(), 
          mergeSequenceEnumerator = mergeSequence.GetEnumerator()) 
    { 
     int i = 1; 
     while (sourceEnumerator.MoveNext()) 
     { 
      yield return sourceEnumerator.Current; 
      i++; 
      if (i == ratio) 
      { 
       if (!mergeSequenceEnumerator.MoveNext()) 
       { 
        mergeSequenceEnumerator.Reset(); 
        mergeSequenceEnumerator.MoveNext(); 
       } 
       yield return mergeSequenceEnumerator.Current; 
       i = 1; 
      } 
     } 
    } 
} 

加上mergeSequenceEnumerator.MoveNext()一些检查，ratio输入value`等

Answer 3

这是我实现：

public static IEnumerable<T> MergeWithRatio<T>(this IEnumerable<T> source, IEnumerable<T> mergeSequence, int ratio) 
{ 
    if (source == null) 
    { 
     throw new ArgumentNullException("source"); 
    } 

    if (mergeSequence == null) 
    { 
     throw new ArgumentNullException("mergeSequence"); 
    } 

    if (ratio <= 1) 
    { 
     throw new ArgumentOutOfRangeException("ratio must be greater one."); 
    } 

    return MergeWithRatioImpl(source, mergeSequence, ratio); 
} 

private static IEnumerable<T> MergeWithRatioImpl<T>(this IEnumerable<T> source, IEnumerable<T> mergeSequence, int ratio) 
{ 
    bool mergeSequenceContainsElements = true; 
    int i = 1; 
    ratio--; 

    using (var sourceEnumerator = source.GetEnumerator()) 
    using (var mergeSequenceEnumerator = mergeSequence.GetEnumerator()) 
    { 
     while (sourceEnumerator.MoveNext()) 
     { 
      yield return sourceEnumerator.Current; 

      if (i++ % ratio == 0) 
      { 
       if (!mergeSequenceEnumerator.MoveNext()) 
       { 
        // ToDo: Should we cache the current values for the case the 
        //  enumerator can't be reset? 
        mergeSequenceEnumerator.Reset(); 
        mergeSequenceContainsElements = mergeSequenceEnumerator.MoveNext(); 
       } 

       if (mergeSequenceContainsElements) 
       { 
        yield return mergeSequenceEnumerator.Current; 
       } 
      } 
     } 
    } 
}