轉載自：http://blog.csdn.net/LYHVOYAGE/article/details/22989659

set集合容器實現了紅黑樹（Red-Black Tree）的平衡二叉檢索樹的的數據結構， 在插入元素時，它會自動調整二叉樹的排列，把該元素放到適當的位置，以確保每個子樹根節點的鍵值大于左子樹所有節點的鍵值，而小于右子樹所有節點的鍵值； 另外，還得確保根節點的左子樹的高度與有字數的高度相等，這樣，二叉樹的高度最小，從而檢索速度最快。要注意的是，它不會重復插入相同鍵值的元素，而采取 忽略處理。

? ? ? ? 平衡二叉檢索樹的檢索使用中序遍歷算法，檢索效率高于vector、deque、和list的容器。另外，采用中序遍歷算法可將鍵值由小到大遍歷出來，所以，可以理解為平衡二叉檢索樹在插入元素時，就會自動將元素按鍵值從小到大的順序排列。

? ? ? ? 構造set集合的主要目的是為了快速檢索，使用set前，需要在程序頭文件中包含聲明“#include<set>”。

1.創建set集合對象

? ? ? ? ? ?創建set對象時，需要指定元素的類型，這一點和其他容器一樣。

[cpp] view plaincopy

#include<iostream>??

#include<set>??

using?namespace?std;??

int?main()??

{??

????set<int>?s;??

????return?0;??

}??

2.元素的插入與中序遍歷

? ? ? ? 采用inset()方法把元素插入到集合中，插入規則在默認的比較規則下，是按元素值從小到大插入，如果自己指定了比較規則函數，則按自定義比較規則函數插入。使用前向迭代器對集合中序遍歷，結果正好是元素排序后的結果。

[cpp] view plaincopy

#include<iostream>??

#include<set>??

using?namespace?std;??

int?main()??

{??

????set<int>?s;??

????s.insert(5);?//第一次插入5，可以插入??

????s.insert(1);??

????s.insert(6);??

????s.insert(3);??

????s.insert(5);?//第二次插入5，重復元素，不會插入??

????set<int>::iterator?it;?//定義前向迭代器??

????//中序遍歷集合中的所有元素??

????for(it?=?s.begin();?it?!=?s.end();?it++)??

????{??

????????cout?<<?*it?<<?"?";??

????}??

????cout?<<?endl;??

????return?0;??

}??

//運行結果：1?3?5?6??

3.元素的方向遍歷

? ? ? ? 使用反向迭代器reverse_iterator可以反向遍歷集合，輸出的結果正好是集合元素的反向排序結果。它需要用到rbegin()和rend()兩個方法，它們分別給出了反向遍歷的開始位置和結束位置。

[cpp] view plaincopy

#include<iostream>??

#include<set>??

using?namespace?std;??

int?main()??

{??

????set<int>?s;??

????s.insert(5);?//第一次插入5，可以插入??

????s.insert(1);??

????s.insert(6);??

????s.insert(3);??

????s.insert(5);?//第二次插入5，重復元素，不會插入??

????set<int>::reverse_iterator?rit;?//定義反向迭代器??

????//反向遍歷集合中的所有元素??

????for(rit?=?s.rbegin();?rit?!=?s.rend();?rit++)??

????{??

????????cout?<<?*rit?<<?"?";??

????}??

????cout?<<?endl;??

????return?0;??

}??

//運行結果：6?5?3?1??

4.元素的刪除

? ? ? ? 與插入元素的處理一樣，集合具有高效的刪除處理功能，并自動重新調整內部的紅黑樹的平衡。刪除的對象可以是某個迭代器位置上的元素、等于某鍵值的元素、一個區間上的元素和清空集合。

[cpp] view plaincopy

#include<iostream>??

#include<set>??

using?namespace?std;??

int?main()??

{??

????set<int>?s;??

????s.insert(5);?//第一次插入5，可以插入??

????s.insert(1);??

????s.insert(6);??

????s.insert(3);??

????s.insert(5);?//第二次插入5，重復元素，不會插入??

????s.erase(6);?//刪除鍵值為6的元素??

????set<int>::reverse_iterator?rit;?//定義反向迭代器??

????//反向遍歷集合中的所有元素??

????for(rit?=?s.rbegin();?rit?!=?s.rend();?rit++)??

????{??

????????cout?<<?*rit?<<?"?";??

????}??

????cout?<<?endl;???

????set<int>::iterator?it;??

??

????it?=?s.begin();??

????for(int?i?=?0;?i?<?2;?i++)??

????????it?=?s.erase(it);???

????for(it?=?s.begin();?it?!=?s.end();?it++)??

????????cout?<<?*it?<<?"?";??

????cout?<<?endl;??

??

????s.clear();??

????cout?<<?s.size()?<<?endl;??

??

????return?0;??

}??

/*?

運行結果：?

5?3?1?

5?

0?????

*/??

5.元素的檢索

? ? ? ? ? 使用find()方法對集合進行檢索，如果找到查找的的鍵值，則返回該鍵值的迭代器位置；否則，返回集合最后一個元素后面的一個位置，即end()。

[cpp] view plaincopy

#include<iostream>??

#include<set>??

using?namespace?std;??

int?main()??

{??

????set<int>?s;??

????s.insert(5);?//第一次插入5，可以插入??

????s.insert(1);??

????s.insert(6);??

????s.insert(3);??

????s.insert(5);?//第二次插入5，重復元素，不會插入??

????set<int>::iterator?it;??

????it?=?s.find(6);?//查找鍵值為6的元素??

????if(it?!=?s.end())??

????????cout?<<?*it?<<?endl;??

????else??

????????cout?<<?"not?find?it"?<<?endl;??

????it?=?s.find(20);??

????if(it?!=?s.end())??

????????cout?<<?*it?<<?endl;??

????else??

????????cout?<<?"not?find?it"?<<?endl;??

????return?0;??

}??

/*?

運行結果：?

6?

not?find?it????

*/??

下面這種方法也能判斷一個數是否在集合中：

[cpp] view plaincopy

#include?<cstdio>??

#include?<set>??

using?namespace?std;??

int?main()?{??

????set?<int>?s;??

????int?a;??

????for(int?i?=?0;?i?<?10;?i++)??

????????s.insert(i);??

????for(int?i?=?0;?i?<?5;?i++)?{??

????????scanf("%d",?&a);??

????????if(!s.count(a))?//不存在??

????????????printf("does?not?exist\n");??

????????else??

????????????printf("exist\n");??

????}??

????return?0;??

}??

6.自定義比較函數

? ? ? ? ?使用insert將元素插入到集合中去的時候，集合會根據設定的比較函數獎該元素放到該放的節點上去。在定義集合的時候，如果沒有指定比較函數，那么采用默認的比較函數，即按鍵值從小到大的順序插入元素。但在很多情況下，需要自己編寫比較函數。

編寫比較函數有兩種方法。

(1)如果元素不是結構體，那么可以編寫比較函數。下面的程序比較規則為按鍵值從大到小的順序插入到集合中。

[cpp] view plaincopy

#include<iostream>??

#include<set>??

using?namespace?std;??

struct?mycomp??

{?//自定義比較函數，重載“（）”操作符??

????bool?operator()?(const?int?&a,?const?int?&b)??

????{??

????????if(a?!=?b)??

????????????return?a?>?b;??

????????else??

????????????return?a?>?b;??

????}??

};??

int?main()??

{??

????set<int,?mycomp>?s;?//采用比較函數mycomp??

????s.insert(5);?//第一次插入5，可以插入??

????s.insert(1);??

????s.insert(6);??

????s.insert(3);??

????s.insert(5);?//第二次插入5，重復元素，不會插入??

????set<int,mycomp>::iterator?it;??

????for(it?=?s.begin();?it?!=?s.end();?it++)??

????????cout?<<?*it?<<?"?";??

????cout?<<?endl;??

????return?0;??

}??

/*?

運行結果：6?5?3?1???

*/??

(2)如果元素是結構體，那么可以直接把比較函數寫在結構體內。

[cpp] view plaincopy

#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?

*/?

轉載于:https://www.cnblogs.com/PrayG/p/5719600.html

總結

以上是生活随笔為你收集整理的STL -set的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。