STL中set的用法

set，顾名思义，就是数学上的集合——每个元素最多只出现一次，并且set中的元素已经从小到大排好序。

 

头文件：#include<set>

常用的函数：

begin()     　　 返回set容器的第一个元素的地址

end() 　　　　 返回set容器的最后一个元素地址

clear()   　　     删除set容器中的所有的元素

empty() 　　　 判断set容器是否为空

max_size() 　   返回set容器可能包含的元素最大个数

size() 　　　　 返回当前set容器中的元素个数

erase(it)             删除迭代器指针it处元素

样例如下：

#include <iostream>
#include <set>

using namespace std;

int main()
{
    set<int> s;
    s.insert(1);
    s.insert(2);
    s.insert(3);
    s.insert(1);
    cout<<"set 的 size 值为 ："<<s.size()<<endl;
    cout<<"set 的 maxsize的值为 ："<<s.max_size()<<endl;
    cout<<"set 中的第一个元素是 ："<<*s.begin()<<endl;
    cout<<"set 中的最后一个元素是:"<<*s.end()<<endl;
    s.clear();
    if(s.empty())
    {
        cout<<"set 为空 ！！！"<<endl;
    }
    cout<<"set 的 size 值为 ："<<s.size()<<endl;
    cout<<"set 的 maxsize的值为 ："<<s.max_size()<<endl;
    return 0;
}

 

还有两个功能类似的函数：count()和find()

1.count() :用来查找set中某个元素出现的次数。这个函数在set并不是很实用，因为一个键值在set只可能出现0或1次，这样就变成了判断某一键值是否在set出现过了。

2.find():   用来查找set中某个元素出现的位置。如果找到，就返回这个元素的迭代器，如果这个元素不存在，则返回 s.end() 。 (最后一个元素的下一个位置，s为set的变量名)

 

样例如下：

#include <iostream>
#include <set>

using namespace std;

int main()
{
    set<int> s;
    set<int>::iterator it;        //创建一个他对应的迭代器

    s.insert(1);
    s.insert(2);
    s.insert(3);
    s.insert(1);
    cout<<"set 中 1 出现的次数是 ："<<s.count(1)<<endl;
    cout<<"set 中 4 出现的次数是 ："<<s.count(4)<<endl;

     it1 = st1.find(4);            //查找数据
    if (it1 != st1.end())        //如果找到就输出数据
    {
        cout <<  *it1 << endl;
    }

    return 0;
}

 

set的遍历，需要用到迭代器，具体的方法见下面的代码：

##include <iostream>
#include<set>
using namespace std;


int main()
{
    set<int> s;                //创建一个int类型的set
 
    s.insert(10);                //插入数据
    s.insert(30);
    s.insert(20);
    s.insert(40);                

    //遍历数据，用迭代器遍历数据
    for (set<int>::iterator it = s.begin(); it != s.end(); ++it)    
    {
        cout << *it << endl;
    }
    //这里用到了set中的元素已经从小到大排好序的性质
    
    return 0;
}

 

最后，不得不提的就是结构体类型（struct ）的set ，使用时必须要重载 '<' 运算符

 

#include<iostream>
#include<set>
#include<string>
using namespace std;
struct Info
{
    string name;
    double score;
    bool operator < (const Info &a) const // 重载“<”操作符，自定义排序规则
    {
        //按score由大到小排序。如果要由小到大排序，使用“>”即可。
        return a.score < score;
    }
};
int main()
{
    set<Info> s;
    Info info;

    //插入三个元素
    info.name = "Jack";
    info.score = 80;
    s.insert(info);
    info.name = "Tom";
    info.score = 99;
    s.insert(info);
    info.name = "Steaven";
    info.score = 60;
    s.insert(info);

    set<Info>::iterator it;
    for(it = s.begin(); it != s.end(); it++)
        cout << (*it).name << " : " << (*it).score << endl; 
    return 0;
}
/*
运行结果：
Tom : 99
Jack : 80
Steaven : 60
*/