常见查找算法
- find // 查找元素
- find_if // 按条件查找元素
- adjacent_find // 查找相邻重复元素
- binary_search // 二分查找法:效率最高的。
- count // 统计元素个数
- count_if // 按条件统计元素个数
find
- 查找指定元素,找到返回指定元素的迭代器,找不到返回结束迭代器end()。
函数原型:
- find(iterator beg, iterator end, value);
- // 按值查找元素,找到返回指定位置迭代器,找不到返回结束迭代器位置
- // beg 开始迭代器
- // end 结束迭代器
- // value 查找的元素
找内置数据类型:
void test01()
{
vector<int> v;
v.push_back(10);
v.push_back(30);
v.push_back(50);
v.push_back(20);
vector<int>::iterator it = find(v.begin(), v.end(), 20);
if (it == v.end())
{
cout << "未找到改元素。" << endl;
}
else
{
cout << "找到该元素:" << *it << endl;
}
}
找自定义数据类型:
参照上面写:发现效果不对:
class Person
{
public:
Person(string name, int age)
{
this->m_Name = name;
this->m_Age = age;
}
string m_Name;
int m_Age;
};
void test02()
{
vector<Person> v;
Person p1("aaa", 8);
Person p2("bbb", 18);
Person p3("ccc", 38);
Person p4("ddd", 28);
v.push_back(p1);
v.push_back(p2);
v.push_back(p3);
v.push_back(p4);
vector<Person>::iterator it = find(v.begin(), v.end(), p2);
if (it == v.end())
{
cout << "未找到改元素。" << endl;
}
else
{
cout << "找到该元素:" << it->m_Name << endl;
}
}
因为:编译没有关于自定义类型如何比较两个自定义类型的方法。我们在find中只输入了p2。但是编译器并不知道,如何比较对象与p2是否相等。
所以需要再Person类中重载==运算符。
class Person
{
public:
Person(string name, int age)
{
this->m_Name = name;
this->m_Age = age;
}
bool operator==(const Person& p)
{
if (p.m_Name == this->m_Name && p.m_Age == this->m_Age)
{
return true;
}
else
{
return false;
}
}
string m_Name;
int m_Age;
};
通过重载==运算符让find知道如何去比较两个自定义类型。
void test02()
{
vector<Person> v;
Person p1("aaa", 8);
Person p2("bbb", 18);
Person p3("ccc", 38);
Person p4("ddd", 28);
v.push_back(p1);
v.push_back(p2);
v.push_back(p3);
v.push_back(p4);
Person pp("bbb", 18);
vector<Person>::iterator it = find(v.begin(), v.end(), pp);
if (it == v.end())
{
cout << "未找到改元素。" << endl;
}
else
{
cout << "找到该元素:" << it->m_Name << endl;
}
}
find_if:按图索骥
- 按条件查找元素
函数原型:
- find_if(iterator beg, iterator end, _func);
- // 按值查找元素,找到返回指定位置迭代器,找不到返回结束迭代器位置
- // beg 开始迭代器
- // end 结束迭代器
- // _Pred 函数或者谓词(返回bool类型的仿函数)
1、查找内置数据类型。
class GreaterFive
{
public:
bool operator()(int val)
{
if (val>5)
{
return true;
}
else
{
return false;
}
}
};
void test01()
{
vector<int> v;
for (int i = 0; i < 10; i++)
{
v.push_back(i);
}
vector<int>::iterator it = find_if(v.begin(), v.end(), GreaterFive());
if (it == v.end())
{
cout << "未找到该元素。" << endl;
}
else
{
cout << "该元素为:" << *it << endl;
}
}
2、查找自定义数据类型。
class Person
{
public:
Person(string name, int age)
{
this->m_Name = name;
this->m_Age = age;
}
string m_Name;
int m_Age;
};
class AgeGreater20
{
public:
bool operator()(const Person& p)
{
if (p.m_Age > 20)
{
return true;
}
else { return false; }
}
};
void test02()
{
vector<Person> v;
Person p1("aaa", 18);
Person p2("bbb", 28);
Person p3("ccc", 38);
Person p4("ddd", 28);
v.push_back(p1);
v.push_back(p2);
v.push_back(p3);
v.push_back(p4);
vector<Person>::iterator it = find_if(v.begin(), v.end(), AgeGreater20());
if (it == v.end())
{
cout << "未找到这个元素" << endl;
}
else
{
cout << "第一个符合条件的人是:" << it->m_Name << endl;
}
}
总结:find_if按条件查找使查找更加灵活,提供的仿函数可以改变不同的策略
adjacent_find:查找相邻重复元素。
功能描述:
- 查找相邻重复元素。
函数原型:
- adjacent_find(iterator beg, iterator end);
- // 查找相邻重复元素,返回相邻元素的第一个位置的迭代器
- // beg 开始迭代器
- // end 结束迭代器
#include <algorithm>
#include <vector>
void test01()
{
vector<int> v;
v.push_back(1);
v.push_back(2);
v.push_back(5);
v.push_back(2);
v.push_back(4);
v.push_back(4);
v.push_back(3);
//查找相邻重复元素
vector<int>::iterator it = adjacent_find(v.begin(), v.end());
if (it == v.end()) {
cout << "找不到!" << endl;
}
else {
cout << "找到相邻重复元素为:" << *it << endl;
}
}
总结:面试题中如果出现查找相邻重复元素,记得用STL中的adjacent_find算法。
binary_search:
功能描述:
- 查找指定元素是否存在
函数原型:
- bool binary_search(iterator beg, iterator end, value);
- // 查找指定的元素,查到 返回true 否则false
- // 注意: 在无序序列中不可用
- // beg 开始迭代器
- // end 结束迭代器
- // value 查找的元素
void test01()
{
vector<int>v;
for (int i = 0; i < 10; i++)
{
v.push_back(i);
}
//二分查找
bool ret = binary_search(v.begin(), v.end(),2);
if (ret)
{
cout << "找到了" << endl;
}
else
{
cout << "未找到" << endl;
}
}
总结:二分查找法查找效率很高,值得注意的是查找的容器中元素必须的有序序列
count
函数原型:
- count(iterator beg, iterator end, value)
- // 统计元素出现次数
- // beg 开始迭代器
- // end 结束迭代器
- // value 统计的元素
//自定义数据类型
class Person
{
public:
Person(string name, int age)
{
this->m_Name = name;
this->m_Age = age;
}
bool operator==(const Person & p)
{
if (this->m_Age == p.m_Age)
{
return true;
}
else
{
return false;
}
}
string m_Name;
int m_Age;
};
void test02()
{
vector<Person> v;
Person p1("刘备", 35);
Person p2("关羽", 35);
Person p3("张飞", 35);
Person p4("赵云", 30);
Person p5("曹操", 25);
v.push_back(p1);
v.push_back(p2);
v.push_back(p3);
v.push_back(p4);
v.push_back(p5);
Person p("诸葛亮",35);
int num = count(v.begin(), v.end(), p);
cout << "num = " << num << endl;
}
总结: 统计自定义数据类型时候,需要配合重载 operator==
count_if
功能描述:
- 按条件统计元素个数
函数原型
- count_if(iterator beg, iterator end, _Pred);
- // 按条件统计元素出现次数
- // beg 开始迭代器
- // end 结束迭代器
- // _Pred 谓词
总结:按值统计用count,按条件统计用count_if。