高效的 Linq Enumerable 的 'Count() == 1'测试

Question

与此 问题 类似，但针对 Linq 进行了改写：

您可以使用 Enumerable;.Any() 测试可枚举是否包含数据。但是，测试枚举是否包含单个值（即 Enumerable.Count() == 1）或大于单个值（即 Enumerable）的有效方法是什么？ .Count() > 1) 而不使用昂贵的计数操作？

Answer 1

int constrainedCount = yourSequence.Take(2).Count();

// if constrainedCount == 0 then the sequence is empty
// if constrainedCount == 1 then the sequence contains a single element
// if constrainedCount == 2 then the sequence has more than one element

Answer 2

一种方法是编写一个新的扩展方法

public static bool IsSingle<T>(this IEnumerable<T> enumerable) {
  using (var enumerator = enumerable.GetEnumerator()) {
    if (!enumerator.MoveNext()) {
      return false;
    }
    return !enumerator.MoveNext();
  }
}

Answer 3

此代码采用了 LukeH 的出色答案，并将其包装为 IEnumerable 扩展，以便您的代码可以处理 None、One 和 >许多而不是0、1和2。

public enum Multiplicity
{
    None,
    One,
    Many,
}

在静态类中，例如 EnumerableExtensions：

public static Multiplicity Multiplicity<TElement>(this IEnumerable<TElement> @this)
{
    switch (@this.Take(2).Count())
    {
        case 0: return General.Multiplicity.None;
        case 1: return General.Multiplicity.One;
        case 2: return General.Multiplicity.Many;
        default: throw new Exception("WTF‽");
    }
}

Answer 4

另一种方式：

bool containsMoreThanOneElement = yourSequence.Skip(1).Any();

或者对于 1 个元素：

bool containsOneElement = yourSequence.Any() && !yourSequence.Skip(1).Any();

Answer 5

高效的 Count() == n 测试：

public static bool CountIsEqualTo<T>(this IEnumerable<T> enumerable, int c) 
{
    using (var enumerator = enumerable.GetEnumerator()) 
    {
        for(var i = 0; i < c ; i++)
            if (!enumerator.MoveNext()) 
                return false;
        
        return !enumerator.MoveNext();
    }
}

编辑，11 年后：
现在我可能会把它写成它

public static bool CountIsEqualTo<T>(this IEnumerable<T> enumerable, int c) 
{
    return enumerable.Take(c + 1).Count() == c;
}

是已接受答案的通用版本。

Answer 6

对于对象的 linq，如果有多个元素，SingleOrDefault 就会抛出异常，因此如果您自己推出一个元素，情况可能会更好。

编辑：现在我已经看到了LukeH的答案，我不得不说我更喜欢它。真希望我自己也能想到这一点！

Answer 7

bool hasTwo = yourSequence.ElementAtOrDefault(1) != default(T);

...如果类的值可以为空，这可能对我们有用。