STL容器中统计算法count, count_if
- 💢功能
- 💢非自定义类型举例
- 💢自定义类型举例
💢功能
- 🥝count用于统计容器中某个特定元素的个数,使用的是==运算符。
- 🥝count_if根据指定的条件统计满足条件的元素的个数。
- 🥝如果统计自定义类型,需要重载==运算符或者自定义pred。
💢非自定义类型举例
👉 👉 👉
count(vec1.begin(), vec1.end(), 5),统计vec1中5的数量。
count_if(vec1.begin(), vec1.end(), [](int val) -> bool {return val > 3;}),统计vec1中满足大于3的元素的数量。
code:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;template<typename T>
void print_vector(const vector<T>& vec)
{for (auto i_vec : vec){cout << i_vec << " ";}cout << endl;
}void test01()
{vector<int> vec1 {1, 2, 5, 7, 25, 5, 5, 7, 33, 44};print_vector(vec1);int cnt = count(vec1.begin(), vec1.end(), 5);cout << "count(vec1.begin(), vec1.end(), 5): " << cnt << endl;
}void test02()
{vector<int> vec1 {1, 2, 5, 7, 25, 5, 5, 7, 33, 44};print_vector(vec1);// count_if举例,记录vector中大于3的元素个数int cnt = count_if(vec1.begin(), vec1.end(), [](int val) -> bool {return val > 3;});cout << "(vec1.begin(), vec1.end(), [](int val) -> bool {val > 10;}): " << cnt << endl;
}void main()
{test01();test02();system("pause");
}result:
1 2 5 7 25 5 5 7 33 44
count(vec1.begin(), vec1.end(), 5): 3
1 2 5 7 25 5 5 7 33 44
(vec1.begin(), vec1.end(), [](int val) -> bool {val > 10;}): 8
💢自定义类型举例
👉 👉 👉
自定义类型在使用时,要重载==运算符。
另外,pred也需要重新实现。
code:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;class Person
{
public:Person(string name, int age) : m_name(name), m_age(age) {}bool operator==(const Person& per){if ((per.m_name == this->m_name) && (per.m_age == this->m_age)){return true;}else{return false;}}
public:string m_name;int m_age;
};class Greater10
{
public:bool operator()(Person & per){return per.m_age > 10;}
};
void print_vector(const vector<Person> &vec)
{for (auto i_vec : vec){cout << "name: " << i_vec.m_name << ", age: " << i_vec.m_age << endl;}cout << endl;
}void test01()
{Person p1("张三", 10);Person p2("李四", 11);Person p3("王五", 12);Person p4("赵六", 10);Person p5("王五", 12);vector<Person> vec1;vec1.push_back(p1);vec1.push_back(p2);vec1.push_back(p3);vec1.push_back(p4);vec1.push_back(p5);print_vector(vec1);int cnt = count(vec1.begin(), vec1.end(), p5);cout << "王五的个数: " << cnt << endl;cnt = count_if(vec1.begin(), vec1.end(), Greater10());cout << "年龄大于10的个数: " << cnt << endl;
}void main()
{test01();system("pause");
}result:
name: 张三, age: 10
name: 李四, age: 11
name: 王五, age: 12
name: 赵六, age: 10
name: 王五, age: 12王五的个数: 2
年龄大于10的个数: 3