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
*/