C++ STL set::find的用法

最新推荐文章于 2024-03-22 19:57:00 发布

cupidove

最新推荐文章于 2024-03-22 19:57:00 发布

阅读量1.8k

点赞数

分类专栏： vc

vc 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

本文介绍了 C++ STL 中 set 的实现原理，通过自定义排序准则，实现 Student 结构体在 set 中的查找操作。重点讨论了如何在 set 中查找 name 和 age 相匹配的 Student 对象。

摘要由CSDN通过智能技术生成

#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;
}

上次面阿里巴巴。面试官问了我这样一个问题，“C++ STL中的set是如何实现的”。当时只答了二叉树，回来查下书，原来一般是红黑树，后悔没好好记住啊。。。

接着，面试官又考了我一道这样的编程题：定义一个Student结构体，包括name和age等数据，要求编程实习在set中查找一个name == "张三"， age == 13的操作。

本来set自己用得不多，当时一下懵了。回来查阅《C++标准程序库》这本书，自己试着实现了下。