Optimize LINQ Count() > X

本文关键字：gt LINQ Count Optimize | 更新日期: 2023-09-27 18:15:53

问题:给定IEnumerable<>，如何检查哪个序列包含的项目多于x ?

MCVE: 

static void Main(string[] args)
{
    var test = Test().Where(o => o > 2 && o < 6); // ToList()
    if (test.Count() > 1) // how to optimize this?
        foreach (var t in test) // consumer
            Console.WriteLine(t);
}
static IEnumerable<int> Test()
{
    for (int i = 0; i < 10; i++)
        yield return i;
}

这里的问题是什么Count()将运行完整的序列，这是1E6 +项目(ToList()也是坏主意)。我也不允许更改消费者代码(这是一个接受完整序列的方法)。

Optimize LINQ Count() > X

对于大的 test集合(当Count()是昂贵的)，您可以尝试一个典型的技巧:

if (test.Skip(1).Any())

在一般情况下 test.Count() > x可以重写为

if (test.Skip(x).Any())

编辑:您可能想要在方法中隐藏这样的技巧，例如EnsureCount:

  public static partial class EnumerableExtensions {
    public static IEnumerable<T> EnsureCount<T>(this IEnumerable<T> source, int count) {
      if (null == source)
        throw new ArgumentNullException("source");
      if (count <= 0)
        foreach (var item in source)
          yield return item;
      else {
        List<T> buffer = new List<T>(count);
        foreach (var item in source) {
          if (buffer == null)
            yield return item;
          else {
            buffer.Add(item);
            if (buffer.Count >= count) {
              foreach (var x in buffer)
                yield return x;
              buffer = null;
            }
          }
        }
      }
    }
  }
所以你的代码就是 
  var test = Test()
    .Where(o => o > 2 && o < 6)
    .EnsureCount(2); // <- Count() > 1, so at least 2 items
  foreach (var t in test)
    Console.WriteLine(t);
相关文章：