c# – 有没有一种简单的方法可以使用LINQ合并两个有序序列?

特定

IEnumerable<T> first;
IEnumerable<T> second;

并且第一和第二都由比较器Func< T,T,int>排序.返回0表示相等,-1表示第一个为“较小”,1表示第二个为“较小”.

是否有一种直接的方式使用LINQ来合并两个序列,使得结果序列也由相同的比较器排序?

我们目前正在使用手工制作的算法,但是直接LINQ语句的可读性会更好.

最佳答案
您可以为此定义扩展方法.就像是

public static IEnumerable<T> MergeSorted<T>(this IEnumerable<T> first, IEnumerable<T> second, Func<T, T, int> comparer) 
{
    using (var firstEnumerator = first.GetEnumerator())
    using (var secondEnumerator = second.GetEnumerator())
    {

        var elementsLeftInFirst = firstEnumerator.MoveNext();
        var elementsLeftInSecond = secondEnumerator.MoveNext();
        while (elementsLeftInFirst || elementsLeftInSecond)
        {
            if (!elementsLeftInFirst)
            {
                    do
                    {
                        yield return secondEnumerator.Current;
                    } while (secondEnumerator.MoveNext());
                    yield break;
            }

            if (!elementsLeftInSecond)
            {
                    do
                    {
                        yield return firstEnumerator.Current;
                    } while (firstEnumerator.MoveNext());
                    yield break;
            }

            if (comparer(firstEnumerator.Current, secondEnumerator.Current) < 0)
            {
                yield return firstEnumerator.Current;
                elementsLeftInFirst = firstEnumerator.MoveNext();
            }
            else
            {
                yield return secondEnumerator.Current;
                elementsLeftInSecond = secondEnumerator.MoveNext();
            }
        }
    }
}

用法:

var s1 = new[] { 1, 3, 5, 7, 9 };
var s2 = new[] { 2, 4, 6, 6, 6, 8 };

var merged = s1.MergeSorted(s2, (a, b) => a > b ? 1 : -1).ToList();

Console.WriteLine(string.Join(", ", merged));

输出:

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

转载注明原文:c# – 有没有一种简单的方法可以使用LINQ合并两个有序序列? - 代码日志