c# - 使用 LINQ 交错多个（超过 2 个）不规则列表

Question

说我有以下数据

IEnumerable<IEnumerable<int>> items = new IEnumerable<int>[] { 
    new int[] { 1, 2, 3, 4 },
    new int[] { 5, 6 },
    new int[] { 7, 8, 9 }
};

返回带有交错项目的平面列表的最简单方法是什么，所以我会得到结果：

1、5、7、2、6、8、3、9、4

注意：内部列表的数量在运行时是未知的。

Answer 1

您所描述的本质上是一种转置方法，其中包含悬垂的项目并且结果是flattened。这是我的尝试：

static IEnumerable<IEnumerable<T>> TransposeOverhanging<T>(
    this IEnumerable<IEnumerable<T>> source)
{
    var enumerators = source.Select(e => e.GetEnumerator()).ToArray();
    try
    {
        T[] g;
        do
        {
            yield return g = enumerators
                .Where(e => e.MoveNext()).Select(e => e.Current).ToArray();
        }
        while (g.Any());
    }
    finally
    {
        Array.ForEach(enumerators, e => e.Dispose());
    }
}

例子：

var result = items.TransposeOverhanging().SelectMany(g => g).ToList();
// result == { 1, 5, 7, 2, 6, 8, 3, 9, 4 }

Answer 2

下面的解决方案非常简单。事实证明，它的速度也几乎是 dtb 提出的解决方案的两倍。

private static IEnumerable<T> Interleave<T>(this IEnumerable<IEnumerable<T>> source )
{
    var queues = source.Select(x => new Queue<T>(x)).ToList();
    while (queues.Any(x => x.Any())) {
        foreach (var queue in queues.Where(x => x.Any())) {
            yield return queue.Dequeue();
        }
    }
}

Answer 3

这是我的尝试，基于dtb 的回答。它避免了外部SelectMany和内部ToArray调用。

public static IEnumerable<T> Interleave<T>(this IEnumerable<IEnumerable<T>> source)
{
    var enumerators = source.Select(e => e.GetEnumerator()).ToArray();
    try
    {
        bool itemsRemaining;
        do
        {
            itemsRemaining = false;
            foreach (var item in 
                     enumerators.Where(e => e.MoveNext()).Select(e => e.Current))
            {
                yield return item;
                itemsRemaining = true;
            }
        }
        while (itemsRemaining);
    }
    finally
    {
        Array.ForEach(enumerators, e => e.Dispose());
    }
}

Answer 4

• 处置所有枚举器，即使抛出异常
• 急切地评估外部序列，但对内部序列使用惰性评估。

---

public static IEnumerable<T> Interleave<T>(IEnumerable<IEnumerable<T>> sequences)
{
    var enumerators = new List<IEnumerator<T>>();
    try
    {
        // using foreach here ensures that `enumerators` contains all already obtained enumerators, in case of an expection is thrown here.
        // this ensures proper disposing in the end
        foreach(var enumerable in sequences)
        {
            enumerators.Add(enumerable.GetEnumerator());
        }

        var queue = new Queue<IEnumerator<T>>(enumerators); 
        while (queue.Any())
        {
            var enumerator = queue.Dequeue();
            if (enumerator.MoveNext())
            {
                queue.Enqueue(enumerator);
                yield return enumerator.Current;
            }
        }
    }
    finally
    {
        foreach(var enumerator in enumerators)
        {
            enumerator.Dispose();
        }
    }
}

Answer 5

虽然它不像“dtb”的答案那么优雅，但它也可以工作并且它是一个单一的衬里:)

Enumerable.Range(0, items.Max(x => x.Count()))
                      .ToList()
                      .ForEach(x =>
                                   {
                                            items
                                            .Where(lstChosen => lstChosen.Count()-1 >= x)
                                            .Select(lstElm => lstElm.ElementAt(x))
                                            .ToList().ForEach(z => Console.WriteLine(z));
                                   });