上次面阿里巴巴。面试官问了我这样一个问题,“C++ STL中的set是如何实现的”。当时只答了二叉树,回来查下书,原来一般是红黑树,后悔没好好记住啊。。。
接着,面试官又考了我一道这样的编程题:定义一个Student结构体,包括name和age等数据,要求编程实现在set中查找一个name == "张三", age == 13的操作。
本来set自己用得不多,当时一下懵了。回来查阅《C++标准程序库》这本书,自己试着实现了下。
- #include <iostream>
- #include <set>
- using namespace std;
- /*Student结构体*/
- struct Student {
- string name;
- int age;
- string sex;
- };
- /*“仿函数"。为Student set指定排序准则*/
- class studentSortCriterion {
- public:
- bool operator() (const Student &a, const Student &b) const {
- /*先比较名字;若名字相同,则比较年龄。小的返回true*/
- if(a.name < b.name)
- return true;
- else if(a.name == b.name) {
- if(a.age < b.age)
- return true;
- else
- return false;
- } else
- return false;
- }
- };
- int main()
- {
- set<Student, studentSortCriterion> stuSet;
- Student stu1, stu2;
- stu1.name = "张三";
- stu1.age = 13;
- stu1.sex = "male";
- stu2.name = "李四";
- stu2.age = 23;
- stu2.sex = "female";
- stuSet.insert(stu1);
- stuSet.insert(stu2);
- /*构造一个测试的Student,可以看到,即使stuTemp与stu1实际上并不是同一个对象,
- *但当在set中查找时,仍会查找成功。这是因为已定义的studentSortCriterion的缘故。
- */
- Student stuTemp;
- stuTemp.name = "张三";
- stuTemp.age = 13;
- set<Student, studentSortCriterion>::iterator iter;
- iter = stuSet.find(stuTemp);
- if(iter != stuSet.end()) {
- cout << (*iter).name << endl;
- } else {
- cout << "Cannot fine the student!" << endl;
- }
- return 0;
- }
运行结果: