STL之set

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<set>

using namespace std;

/*仿函数*/
class mycompare{
public:
    bool operator()(int v1,int v2){
        return v1>v2;
    }
};

//初始化
void test01(){


    set<int,mycompare> s1;/*自动进行排序，默认从小到大*/
    s1.insert(7);
    s1.insert(2);
    s1.insert(4);
    s1.insert(5);
    s1.insert(1);
    for(set<int>::iterator it=s1.begin(); it!=s1.end(); it++ ){
        cout<<(*it)<<" ";
    }
    cout<<endl;

    //如何改变默认排序？

    /*赋值操作*/
    set<int> s2;
//    s2=s1;

    //删除操作
    s1.erase(s1.begin());
    s1.erase(7);


    for(set<int>::iterator it=s1.begin(); it!=s1.end(); it++ ){
        cout<<(*it)<<" ";
    }
    cout<<endl;
}

void test02(){
    set<int> s1;/*自动进行排序，默认从小到大*/
    s1.insert(7);
    s1.insert(2);
    s1.insert(4);
    s1.insert(5);
    s1.insert(1);
    /*find()函数返回迭代器*/
    set<int>::iterator vec=s1.find(4);
    if(vec==s1.end()){
        cout<<"没有找到！"<<endl;
    }
    else{
        cout<<"vec:"<<(*vec)<<endl;
    }

    /*lower_bound()返回第一个大于等于的迭代器*/
    vec=s1.lower_bound(2);
    if(vec==s1.end()){
        cout<<"没有找到！"<<endl;
    }
    else{
        cout<<"vec:"<<(*vec)<<endl;
    }
    /*找到第一个大于key的值*/
    vec=s1.upper_bound(2);
    if(vec==s1.end()){
        cout<<"没有找到！"<<endl;
    }
    else{
        cout<<"vec:"<<(*vec)<<endl;
    }
    //equal_range 返回lower_bound和upper_bound值,运用pair接收
    pair<set<int>::iterator,set<int>::iterator> myvec=s1.equal_range(2);
    if(myvec.first==s1.end()){
        cout<<"没有找到！"<<endl;
    }
    else{
        cout<<"myvec:" <<*(myvec.first)<<endl;
    }
    if(myvec.second==s1.end()){
        cout<<"没有找到！"<<endl;
    }
    else{
        cout<<"myvec:"<<*(myvec.second)<<endl;
    }
}

//在set中放自定义类型
class Person{
public:
    int id;
    int age;
public:
    Person(int age,int id):id(id),age(age){}
};
//自定义排序规则
class flag{
public:
    bool operator()(Person p1, Person p2){
        return p1.age<p2.age;
    }
};
void test03(){
    set<Person,flag> sp;//set需要排序，当放对象的时候，要提前给定排序规则
    Person p1(10,20),p2(30,40),p3(50,60);
    sp.insert(p1);
    sp.insert(p2);
    sp.insert(p3);

    Person p4(10,30);
    for(set<Person,mycompare>::iterator it =sp.begin(); it!=sp.end(); it++ ){
        cout<<(*it).age<<" "<<(*it).id<<endl;
    }

    //查找
    set<Person,mycompare>::iterator vec=sp.find(p4);
    if(vec==sp.end()){
        cout<<"没有找到！"<<endl;
    }
    else{
        cout<<"找到！"<<(*vec).age<<" "<<(*vec).id<<endl;
    }
}


int main(){
    test01();
    return 0;
}