优化通用 AddDistinct 扩展方法

Question

我可以更有效地编写这些函数吗？

public static void AddDistinct<T>(this ICollection<T> source, params ICollection<T>[] collections)
{
    (from collection in collections from item in collection where !source.Contains(item) select item).ForEach(source.Add);
}

public static void AddDistinct<T>(this ICollection<T> source, Func<T, bool> predicate, params ICollection<T>[] collections)
{
    collections.ForEach(collection => collection.Where(predicate).Where(item => !source.Contains(item)).ForEach(source.Add));
}

public static void AddDistinct<T>(this ICollection<T> source, params T[] items)
{
    items.Where(item => !source.Contains(item)).ForEach(source.Add);
}

public static void AddDistinct<T>(this ICollection<T> source, Func<T, bool> predicate, params T[] items)
{
    items.Where(predicate).Where(x => !source.Contains(x)).ForEach(source.Add);
}

Answer 1

您可以使用 Except 方法：

public static void AddDistinct<T>(this ICollection<T> source, params ICollection<T>[] collections)
{
    var itemsToAdd = collections.SelectMany(x => x).Where(predicate).Except(source).ToArray();
    itemsToAdd.ForEach(source.Add));
}

public static void AddDistinct<T>(this ICollection<T> source, Func<T, bool> predicate, params ICollection<T>[] collections)
{
    var itemsToAdd = collections.SelectMany(x => x).Where(predicate).Except(source).ToArray();
    itemsToAdd.ForEach(source.Add));
}

public static void AddDistinct<T>(this ICollection<T> source, params T[] items)
{
    var itemsToAdd = items.Except(source).ToArray();
    itemsToAdd.ForEach(source.Add));
}

public static void AddDistinct<T>(this ICollection<T> source, Func<T, bool> predicate, params T[] items)
{
    var itemsToAdd = items.Where(predicate).Except(source).ToArray();
    itemsToAdd.ForEach(source.Add));
}

请注意，由于 Where 和 Except 具有延迟和流式执行，因此您需要 ToArray()< /code> 调用以确保在向其中添加任何内容之前完成 source 的枚举（因为在枚举集合时无法对其进行修改）。

Answer 2

是的，使用例如哈希集，您可以比 O(N*M) 做得更好。

Answer 3

我认为使用 HashSet 和 IEnumerable 可以更高效。 HashSet 可以高效地跟踪重复项，您只需浏览一次源即可将它们加载到集合中。除非我弄错了，否则我认为 ICollection 和 T[] 都是 IEnumerable 吗？

public static void AddDistinct<T>(this ICollection<T> source, params IEnumerable<T> items)
{
    var set = new HashSet<T>(source);
    foreach(var item in items)
    {
        if(set.Add(item))
            source.Add(item);
    }
}

但不确定您需要这些谓词版本做什么。您不能在添加项目之前过滤它们吗？