我需要比较两个可能缺少元素的非常大的集合

本文关键字:元素 非常 集合 两个 比较 | 更新日期: 2023-09-27 18:35:06

我在为我的工作编写的程序中遇到了一堵砖墙。您不需要具体了解上下文,但长话短说,我有两个集合,每个集合大约 ~650k 条记录。

假设集合 A 是我知道正确的集合,而集合 B 是我知道不正确的集合。

集合 B 包含一个复杂对象,该对象具有与集合 A 中的元素相同类型的属性(换句话说,它看起来有点像这样):

// Where T : IComparable
IEnumerable<DateTime> A = ...; // Collection of T elements
IEnumerable<Complex> B = ...; // Collection of complex elements.
class Complex<DateTime>
{
   public DateTime Time { get; set; }
   .....
}

我的问题是,我基本上需要按顺序枚举 A,看看 A 的当前元素是否存在于 B 中的 Complex 对象中;如果它不存在,那么我需要创建一个 Complex 对象来封装该元素(以及其他内容)。

当我意识到两个列表都有 650,000 个元素长,大约时,就会出现问题。我无法减少数据集;我必须使用这650,000。现在我已经使用了ICollection.Contains(),并且我尝试了二叉搜索的(天真)实现,但它花费的时间太长了。

你对我有什么建议吗?

编辑:如果有帮助,T实现了IComparable。编辑2:更多背景:IEnumerable 是使用 Linq To Object 从数据表中检索的。

        IEnumerable<Complex> set = set.Tbl
            .Where(dbObject => dbObject.TS.CompareTo(default(DateTime)) != 0)
            .Select(Func<DataRow,Complex>) // Function that wraps the DataRow in a Complex object
            // Just done to make debugging a little easier so we still have a large sample but small enough that it doesn't make me grow a beard
            .Take(100000) 
            .AsEnumerable<Complex>();

为了完整起见,以防此问题被存档并且其他人需要解决此问题,我当前的实现看起来有点像这样

        BDataSet bSet = new BDataSet();
        B_LUTableAdapter adap = new B_LUTableAdapter();
        adap.Fill(bSet.B_LU);
        IEnumerable<Complex> w3 = bSet.B
            .Where(dbObject => dbObject.TS.CompareTo(default(DateTime)) != 0)
            // Function that just wraps datarow into a complex object
            .Select(Func<DataRow, Complex>)
            // Just for sake of debugging speed
            .Take(100000)
            .AsEnumerable<Complex>();
        List<Complex> b = bSet.OrderBy(x => x.Time).ToList<Complex>();
        // Get last & first timestamps
        // Some of the timestamps in b are 01/01/1011 for some reason,
        // So we do this check.
        Complex start = b.Where(x => x.Time != default(DateTime)).First();
        Complex end = b.Last();
        List<DateTime> a = new List<DateTime>();
        // RoundSeconds reduces seconds in a DateTime to 0.
        DateTime current = RoundSeconds(new DateTime(start.Time.Ticks));
        while (current.CompareTo(RoundSeconds(end.Time)) <= 0)
        {
            a.Add(current);
            current = current.Add(TimeSpan.FromMinutes(1));
        }
        IEnumerable<DateTime> times = b.Select(x => x.Time);
        var missing = a.Where(dt => times.Contains(dt));
        foreach (var dt in missing)
        {
            adap.Insert(dt, 0, "", "", "", null, 0, 0);
            // This has since been changed to List.Add()
        }

多亏了Cosmin,这个问题现在已经解决了,完成的实现是这样的: 预期列表 = 新列表(); 日期时间当前 = 圆秒(新的日期时间(开始。时间滴答声));

        while (current.CompareTo(RoundSeconds(end.Time)) <= 0)
        {
            expected.Add(current);
            current = current.Add(TimeSpan.FromMinutes(1));
        }
        Console.WriteLine("Expecting {0} intervals.", expected.Count);
        var missing = b.FindAllMissing(expected, x => x.Time);
        if(!missing.Any()) return;
        Console.WriteLine("{0} missing intervals.", missing.Count());
        foreach (var dt in missing)
        {
            b.Add(new Complex() { /* some values */ });
            //Console.WriteLine("'t> Inserted new record at {0}", dt);
        }
        //.....
        public static IEnumerable<Basic> FindAllMissing<Basic, Complex>(this IEnumerable<Complex> complexList,
        IEnumerable<Basic> basicList,
        Func<Complex, Basic> selector)
        {
            HashSet<Basic> inComplexList = new HashSet<Basic>();
            foreach (Complex c in complexList)
                inComplexList.Add(selector(c));
            List<Basic> missing = new List<Basic>();
            foreach (Basic basic in basicList)
                if (!(inComplexList.Contains(basic)))
                    missing.Add(basic);
            return missing;
        }

我需要比较两个可能缺少元素的非常大的集合

逐步:

  • 使用O(1)通用集合之一创建已在第二个集合中的T的可快速搜索列表。我可以建议HashSet<T>
  • 枚举第二个集合,并从第一步开始将所有T放入集合中。
  • 枚举第一个集合,并检查每个项是否在步骤 1 中创建的集合中。由于该操作O(1)您现在得到了一个O(n)解决方案。
  • 享受。

下面是一个将该算法实现为泛型扩展方法的类,以使其对 LINQ 更加友好。将它的参数作为IEnumerable<T>并返回IEnumerable<T>,对类型(TComplex)没有做出任何假设。在我的测试中,我使用Tuple<int,int>列表作为复杂类型,使用简单int作为简单类型。控制台应用程序用 600000 个值填充List<Tuple<int,int>>,然后将 100000 个值放入简单List<int>中,该使用枚举器对List<Tuple<int,int>>中未找到的所有简单值进行计数;它是如此之快,你没有机会看到它的工作,当你点击F5它只是显示结果。

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace ConsoleApplication2
{
    static class FixProblem
    {
        public static IEnumerable<T> FindAllThatNeedCreating<T, Complex>(this IEnumerable<Complex> list_of_complex, IEnumerable<T> list_of_T, Func<Complex, T> extract)
        {
            HashSet<T> T_in_list_of_complex = new HashSet<T>();
            foreach (Complex c in list_of_complex)
                T_in_list_of_complex.Add(extract(c));
            List<T> answer = new List<T>();
            foreach (T t in list_of_T)
                if (!T_in_list_of_complex.Contains(t))
                    answer.Add(t);
            return answer;
        }
    }
    class Program
    {
        static void Main(string[] args)
        {
            // Test the code
            List<Tuple<int, int>> complex = new List<Tuple<int, int>>();
            List<int> simple = new List<int>();
            // Fill in some random data
            Random rnd = new Random();
            for (int i = 1; i < 600000; i++)
                complex.Add(new Tuple<int, int>(rnd.Next(), rnd.Next()));
            for (int i = 1; i < 100000; i++)
                simple.Add(rnd.Next());
            // This is the magic line of code:
            Console.WriteLine(complex.FindAllThatNeedCreating(simple, x => x.Item1).Count());
            Console.ReadKey();
        }
    }
}

我建议将复杂对象存储在字典中,使用 A 类型属性作为键。然后,您可以非常快速地查看字典中是否存在 A 中的任何元素。每个查找操作将为 O(1),整体性能应为 O(n)。

或者,您可以调用 。ToLookup() 在现有的 IEnumerable 上,并在 lambda 表达式中创建键和值。返回的 ILookup 应该非常适合您的需求(即从 A 中查找值)。这里的另一个优点是,如果您有多个包含 A 的复杂对象,则在使用 Lookup 时将获得它们的集合。

如果我了解您的要求,那么我认为这段代码效果很好。我已经将两个集合的记录提高到 6.5m 记录,并在 11 秒内完成。将其减少到 650k 条记录只需不到一秒钟的时间。

var rnd = new Random();
var xs = Enumerable
    .Range(0, 650000)
    .Select(x => new Complex<int>()
    {
        MyComplex = rnd.Next(0, 100001)
    })
    .ToList();
var ys = Enumerable
    .Range(0, 650000)
    .Select(x => rnd.Next(0, 100001))
    .ToArray();
var xsLookup = xs.ToLookup(x => x.MyComplex);
var newYs = ys.Where(y => !xsLookup[y].Any()).ToArray();
newYs
    .ToList()
    .ForEach(y =>
    {
        xs.Add(new Complex<int>()
        {
            MyComplex = y
        });
    });

更新:首先,停止使用数据集。我建议你使用Linq to SQL或Entity Framework。

试试这个:

        var lookup = B.ToLookup(c => c.MyComplex);
        var noMatch = from a in A
                      where !lookup.Contains(a)
                      select a;

应该更快,但要衡量。

然后尝试

        var lookup = B.AsParallel().ToLookup(c => c.MyComplex);
        var noMatch = from a in A.AsParallel()
                      where !lookup.Contains(a)
                      select a;

并再次测量。

显然,请确保 A 中的对象类型覆盖GetHashCode(),并且Equals(object)相关且高效。特别是GetHashCode()应该很有可能不同的对象具有不同的哈希代码,并且仍然很快。

更新:由于我们现在知道 A 中的对象类型是 DateTime,因此对 GetHashCode()Equals(object) 的要求是可以的。

代码变为

        var lookup = B.ToLookup(c => c.Time);
        var noMatch = from a in A
                      where !lookup.Contains(a)
                      select a;

如果两个列表都按同一键排序,则可以同时遍历两者。这样做,我能够将时间缩短到比containsexcept的时间更少,甚至比我看到的toLookup时间还要短。

var A = SimpleCollection; // source of truth
var B = GetComplexOnDemand().ToArray();
var missing = new List<Complex<DateTime>>();
int cpxIx = 0;
int simpleIx = 0;
while (simpleIx < A.Count)
{
    if (cpxIx >= B.Length)
    {
        missing.Add(new Complex<DateTime>() {InnerValue = A[simpleIx]});
        simpleIx++;
        continue;
    }
    if (A[simpleIx] != B[cpxIx].InnerValue)
    {
        missing.Add(new Complex<DateTime>() {InnerValue = A[simpleIx]});
        simpleIx++;
        continue;
    }
    cpxIx++;
    simpleIx++;
}

为了生成测试数据,我做了以下操作:

private static readonly List<DateTime> SimpleCollection = 
    Enumerable.Range(1, SimpleSize).Select(t => new DateTime(DateTime.Now.Ticks + t)).ToList();
public static IEnumerable<Complex<DateTime>> GetComplexOnDemand()
{
    for (int i = 1; i <= ComplexSize; i+=2)
    {
        yield return new Complex<DateTime>() { InnerValue = SimpleCollection[i] };
    }
}