LINQ 相交，多个列表，一些为空

Question

我正在尝试寻找与 LINQ 的交集。

示例：

List<int> int1 = new List<int>() { 1,2 };
List<int> int2 = new List<int>();
List<int> int3 = new List<int>() { 1 };
List<int> int4 = new List<int>() { 1, 2 };
List<int> int5 = new List<int>() { 1 };

想要返回：1，因为它存在于所有列表中。如果我运行：

var intResult= int1
            .Intersect(int2)
            .Intersect(int3)
            .Intersect(int4)
            .Intersect(int5).ToList();

它什么也不返回，因为 1 显然不在 int2 列表中。无论一个列表是否为空，我如何让它工作？

使用上面的例子或：

List<int> int1 = new List<int>() { 1,2 };
List<int> int2 = new List<int>();
List<int> int3 = new List<int>();
List<int> int4 = new List<int>();
List<int> int5 = new List<int>();

在这种情况下我如何返回 1 和 2.. 我不知道列表是否已填充...

Answer 1

如果您需要一步完成，最简单的解决方案是过滤掉空列表：

public static IEnumerable<T> IntersectNonEmpty<T>(this IEnumerable<IEnumerable<T>> lists)
{
    var nonEmptyLists = lists.Where(l => l.Any());
    return nonEmptyLists.Aggregate((l1, l2) => l1.Intersect(l2));
}

然后您可以在列表集合或其他IEnumerables 上使用它：

IEnumerable<int>[] lists = new[] { l1, l2, l3, l4, l5 };
var intersect = lists.IntersectNonEmpty();

您可能更喜欢常规的静态方法：

public static IEnumerable<T> IntersectNonEmpty<T>(params IEnumerable<T>[] lists)
{
    return lists.IntersectNonEmpty();
}

var intersect = ListsExtensionMethods.IntersectNonEmpty(l1, l2, l3, l4, l5);

Answer 2

您可以编写一个扩展方法来定义该行为。像

static class MyExtensions
{
    public static IEnumerable<T> IntersectAllIfEmpty<T>(this IEnumerable<T> list, IEnumerable<T> other)
    {
        if (other.Any())
            return list.Intersect(other);
        else
            return list;
    }
}

所以下面的代码会打印 1。

List<int> list1 = new List<int>() { 1, 2 };
List<int> list2 = new List<int>();
List<int> list3 = new List<int>() { 1 };

foreach (int i in list1.IntersectAllIfEmpty(list2).IntersectAllIfEmpty(list3))
    Console.WriteLine(i);

更新：

Anon 在 cmets 中提出了一个很好的问题。如果list 本身为空，则上述函数将 生成一个空集，这应该是可取的。这意味着如果方法链中的第一个列表或任何交集的结果集为空，则最终结果将为空。

要允许空的第一个列表但不允许空结果集，您可以采用不同的方法。这是一种方法，它不是扩展方法，而是采用 IEnumerables 的 params 数组并首先过滤掉空集，然后尝试与其余的相交。

public static IEnumerable<T> IntersectAllIfEmpty<T>(params IEnumerable<T>[] lists)
{
    IEnumerable<T> results = null;

    lists = lists.Where(l => l.Any()).ToArray();

    if (lists.Length > 0)
    {
        results = lists[0];

        for (int i = 1; i < lists.Length; i++)
            results = results.Intersect(lists[i]);
    }
    else
    {
        results = new T[0];
    }

    return results;
}

你会这样使用它

List<int> list0 = new List<int>();
List<int> list1 = new List<int>() { 1, 2 };
List<int> list2 = new List<int>() { 1 };
List<int> list3 = new List<int>() { 1,2,3 };

foreach (int i in IntersectAllIfEmpty(list0, list1, list2, list3))
{
    Console.WriteLine(i);
}