std::count, std::count_if

来自cppreference.com
< cpp‎ | algorithm
 
 
算法库
受约束算法及范围上的算法 (C++20)
包含算法例如 ranges::copy, ranges::sort, ...
执行策略 (C++17)
排序和相关操作
划分操作
排序操作
二分搜索操作(在已划分范围上)
集合操作(在有序范围上)
归并操作(在有序范围上)
堆操作
最小/最大操作
(C++11)
(C++17)
字典序比较操作
排列操作
C 库

数值运算
(C++11)                       
在未初始化内存上的操作
 
在标头 <algorithm> 定义
(1)
template< class InputIt, class T >

typename std::iterator_traits<InputIt>::difference_type

    count( InputIt first, InputIt last, const T& value );
(C++20 起为 constexpr)
(C++26 前)
template< class InputIt, class T = typename std::iterator_traits

                                       <InputIt>::value_type >
constexpr typename std::iterator_traits<InputIt>::difference_type

    count( InputIt first, InputIt last, const T& value );
(C++26 起)
(2)
template< class ExecutionPolicy, class ForwardIt, class T >

typename std::iterator_traits<ForwardIt>::difference_type
    count( ExecutionPolicy&& policy,

           ForwardIt first, ForwardIt last, const T& value );
(C++17 起)
(C++26 前)
template< class ExecutionPolicy,

          class ForwardIt, class T = typename std::iterator_traits
                                         <ForwardIt>::value_type >
typename std::iterator_traits<ForwardIt>::difference_type
    count( ExecutionPolicy&& policy,

           ForwardIt first, ForwardIt last, const T& value );
(C++26 起)
template< class InputIt, class UnaryPred >

typename std::iterator_traits<InputIt>::difference_type

    count_if( InputIt first, InputIt last, UnaryPred p );
(3) (C++20 起为 constexpr)
template< class ExecutionPolicy, class ForwardIt, class UnaryPred >

typename std::iterator_traits<ForwardIt>::difference_type
    count_if( ExecutionPolicy&& policy,

              ForwardIt first, ForwardIt last, UnaryPred p );
(4) (C++17 起)

返回范围 [firstlast) 中满足特定判别标准的元素数。

1) 计数等于 value 的元素(使用 operator==)。
3) 计数谓词 p 对其返回 true 的元素。
2,4)(1,3),但按照 policy 执行。
这些重载只有在

std::is_execution_policy_v<std::decay_t<ExecutionPolicy>>

(C++20 前)

std::is_execution_policy_v<std::remove_cvref_t<ExecutionPolicy>>

(C++20 起)
true 时时才会参与重载决议。

参数

first, last - 要检验的元素范围
value - 要搜索的值
policy - 所用的执行策略。细节见执行策略
p - 对要求的元素则返回 ​true 的一元谓词。

对每个(可为 const 的) VT 类型参数 v ,其中 VTInputIt 的值类型,表达式 p(v) 必须可转换到 bool,无关乎值类别,而且必须不修改 v 。从而不允许 VT& 类型参数,亦不允许 VT ,除非对 VT 而言移动等价于复制 (C++11 起)。 ​

类型要求
-
InputIt 必须满足老式输入迭代器 (LegacyInputIterator)
-
ForwardIt 必须满足老式向前迭代器 (LegacyForwardIterator)
-
UnaryPred 必须满足谓词 (Predicate)

返回值

范围 [firstlast) 中满足以下条件的迭代器 it 的个数:

1,2) *it == valuetrue
3,4) p(*it) != falsetrue

复杂度

给定 Nstd::distance(first, last)

1,2) 应用 Noperator==value 进行比较。
3,4) 应用 N 次谓词 p

异常

拥有名为 ExecutionPolicy 的模板形参的重载按下列方式报告错误:

  • 如果作为算法一部分调用的函数的执行抛出异常,且 ExecutionPolicy标准策略之一,那么调用 std::terminate。对于任何其他 ExecutionPolicy,行为由实现定义。
  • 如果算法无法分配内存,那么抛出 std::bad_alloc

注解

对于无任何额外判别标准的 [firstlast) 中的元素数,见 std::distance

功能特性测试 标准 功能特性
__cpp_lib_algorithm_default_value_type 202403 (C++26) 算法中的列表初始化 (1,2)

可能的实现

参阅 countlibstdc++libc++ 中的实现。

参阅 count_iflibstdc++libc++ 中的实现。


count
template<class InputIt, class T = typename std::iterator_traits<InputIt>::value_type>
typename std::iterator_traits<InputIt>::difference_type
    count(InputIt first, InputIt last, const T& value)
{
    typename std::iterator_traits<InputIt>::difference_type ret = 0;
    for (; first != last; ++first)
        if (*first == value)
            ++ret;
    return ret;
}
count_if
template<class InputIt, class UnaryPred>
typename std::iterator_traits<InputIt>::difference_type
    count_if(InputIt first, InputIt last, UnaryPred p)
{
    typename std::iterator_traits<InputIt>::difference_type ret = 0;
    for (; first != last; ++first)
        if (p(*first))
            ++ret;
    return ret;
}

示例

#include <algorithm>
#include <array>
#include <cassert>
#include <complex>
#include <iostream>
#include <iterator>
 
int main()
{
    constexpr std::array v = {1, 2, 3, 4, 4, 3, 7, 8, 9, 10};
    std::cout << "v:";
    std::copy(v.cbegin(), v.cend(), std::ostream_iterator<int>(std::cout, " "));
    std::cout << '\n';
 
    // 确定有多少个整数匹配目标值。
    for (const int target: {3, 4, 5})
    {
        const int num_items = std::count(v.cbegin(), v.cend(), target);
        std::cout << "目标数字:" << target << ",计数:" << num_items << '\n';
    }
 
    // 用 lambda 表达式计量能被 4 整除的元素数。
    int count_div4 = std::count_if(v.begin(), v.end(), [](int i) { return i % 4 == 0; });
    std::cout << "能被 4 整除的数字个数:" << count_div4 << '\n';
 
    // 简化版的 distance,具有 O(N) 复杂度:
    auto distance = [](auto first, auto last)
    {
        return std::count_if(first, last, [](auto){ return true; });
    };
    static_assert(distance(v.begin(), v.end()) == 10);
 
    std::array<std::complex<double>, 3> nums{{{4, 2}, {1, 3}, {4, 2}}};
    #ifdef __cpp_lib_algorithm_default_value_type
        // 推导的 T 使得列表初始化成为可能
        auto c = std::count(nums.cbegin(), nums.cend(), {4, 2});
    #else
        auto c = std::count(nums.cbegin(), nums.cend(), std::complex<double>{4, 2});
    #endif
    assert(c == 2);
}

输出:

v:1 2 3 4 4 3 7 8 9 10
目标数字:3,计数:2
目标数字:4,计数:2
目标数字:5,计数:0
能被 4 整除的数字个数:3

缺陷报告

下列更改行为的缺陷报告追溯地应用于以前出版的 C++ 标准。

缺陷报告 应用于 出版时的行为 正确行为
LWG 283 C++98 T 需要是可相等比较 (EqualityComparable) 的,但是 InputIt 的值类型不一定是 T 移除该要求

参阅

返回两个迭代器间的距离
(函数模板)
返回满足指定判别标准的元素数
(niebloid)