【STL】count_if

功能

返回满足条件的元素个数

模版

template <class InputIterator, class Predicate>
  typename iterator_traits<InputIterator>::difference_type  //返回值
    count_if (InputIterator first, InputIterator last, UnaryPredicate pred); 

实现

template <class InputIterator, class UnaryPredicate>
  typename iterator_traits<InputIterator>::difference_type
    count_if (InputIterator first, InputIterator last, UnaryPredicate pred)
{
  typename iterator_traits<InputIterator>::difference_type ret = 0;
  while (first!=last) 
 {
    if (pred(*first))
         ++ret;
    ++first;
  }
  return ret;
}

参数

1. first, last: 输入迭代器指出首尾的位置，范围是[first, last),即包括第一个而不包括last。
2. pred: 一元函数名字，接收范围内的一个元素作为参数，返回bool值。函数不得修改其参数。可以为函数指针或函数对象。

案例

案例1. pred为bool函数

代码

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
bool IsOdd(int i)
{
    return ((i % 2) == 1);
}
int main()
{
    vector<int> vec;
    for(int i=0; i<10; ++i)
        vec.push_back(i);
    int mycount = count_if(vec.begin(), vec.end(), IsOdd);
    cout << "My vector contains " << mycount << " odd values." << endl;
}

输出

案例2. pred为函数对象

代码

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
using namespace std;
class GT_cls
{
    public:
        GT_cls(int val) : bound(val) {}
        bool operator()(const string &s)
        { return s.size() >= bound; }
    private:
        string::size_type bound;
};
int main()
{
    vector<string> vec;
    vec.push_back("hello1");
    vec.push_back("hello12");
    vec.push_back("hello123");
    vec.push_back("hello1234");
    vec.push_back("hello12345");
    GT_cls tmp(7);  //函数对象比函数更灵活
    cout << count_if(vec.begin(), vec.end(), tmp) << endl;
}

输出

4

复杂度

O(1)

参考

http://www.cplusplus.com/reference/algorithm/count_if/