C++中set用法詳解

轉載自大佬的博客https://blog.csdn.net/Strawberry_595/article/details/81188509

 

1.關於set

C++ STL 之所以得到廣泛的贊譽，也被很多人使用，不只是提供了像vector, string, list等方便的容器，更重要的是STL封裝了許多復雜的數據結構算法和大量常用數據結構操作。vector封裝數組，list封裝了鏈表，map和set封裝了二叉樹等，在封裝這些數據結構的時候，STL按照程序員的使用習慣，以成員函數方式提供的常用操作，如：插入、排序、刪除、查找等。讓用戶在STL使用過程中，並不會感到陌生。

關於set，必須說明的是set關聯式容器。set作為一個容器也是用來存儲同一數據類型的數據類型，並且能從一個數據集合中取出數據，在set中每個元素的值都唯一，而且系統能根據元素的值自動進行排序。應該注意的是set中數元素的值不能直接被改變。C++ STL中標准關聯容器set, multiset, map, multimap內部采用的就是一種非常高效的平衡檢索二叉樹：紅黑樹，也成為RB樹(Red-Black Tree)。RB樹的統計性能要好於一般平衡二叉樹，所以被STL選擇作為了關聯容器的內部結構。

 關於set有下面幾個問題：

（1）為何map和set的插入刪除效率比用其他序列容器高？

大部分人說，很簡單，因為對於關聯容器來說，不需要做內存拷貝和內存移動。說對了，確實如此。set容器內所有元素都是以節點的方式來存儲，其節點結構和鏈表差不多，指向父節點和子節點。結構圖可能如下：


　  A
　  / \
　 B C
　/ \ / \
  D E F G

因此插入的時候只需要稍做變換，把節點的指針指向新的節點就可以了。刪除的時候類似，稍做變換后把指向刪除節點的指針指向其他節點也OK了。這里的一切操作就是指針換來換去，和內存移動沒有關系。

（2）為何每次insert之后，以前保存的iterator不會失效？

iterator這里就相當於指向節點的指針，內存沒有變，指向內存的指針怎么會失效呢(當然被刪除的那個元素本身已經失效了)。相對於vector來說，每一次刪除和插入，指針都有可能失效，調用push_back在尾部插入也是如此。因為為了保證內部數據的連續存放，iterator指向的那塊內存在刪除和插入過程中可能已經被其他內存覆蓋或者內存已經被釋放了。即使時push_back的時候，容器內部空間可能不夠，需要一塊新的更大的內存，只有把以前的內存釋放，申請新的更大的內存，復制已有的數據元素到新的內存，最后把需要插入的元素放到最后，那么以前的內存指針自然就不可用了。特別時在和find等算法在一起使用的時候，牢記這個原則：不要使用過期的iterator。

（3）當數據元素增多時，set的插入和搜索速度變化如何？

如果你知道log2的關系你應該就徹底了解這個答案。在set中查找是使用二分查找，也就是說，如果有16個元素，最多需要比較4次就能找到結果，有32個元素，最多比較5次。那么有10000個呢？最多比較的次數為log10000，最多為14次，如果是20000個元素呢？最多不過15次。看見了吧，當數據量增大一倍的時候，搜索次數只不過多了1次，多了1/14的搜索時間而已。你明白這個道理后，就可以安心往里面放入元素了。

2.set中常用的方法

 

begin()

end()

clear()

empty()

max_size()

size()

rbegin()

rend()

begin()     　　 ,返回set容器的第一個元素
end() 　　　　 ,返回set容器的最后一個元素
clear()   　　     ,刪除set容器中的所有的元素
empty() 　　　,判斷set容器是否為空
max_size() 　 ,返回set容器可能包含的元素最大個數
size() 　　　　 ,返回當前set容器中的元素個數
rbegin　　　　 ,返回的值和end()相同
rend()　　　　 ,返回的值和rbegin()相同

寫一個程序練一練這幾個簡單操作吧：
   

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

 運行結果：

小結：插入3之后雖然插入了一個1，但是我們發現set中最后一個值仍然是3，這就是set 。還要注意begin() 和 end()函數是不檢查set是否為空的，使用前最好使用empty()檢驗一下set是否為空. 

count

count() 用來查找set中某個某個鍵值出現的次數。這個函數在set並不是很實用，因為一個鍵值在set只可能出現0或1次，這樣就變成了判斷某一鍵值是否在set出現過了。

示例代碼：
  

    #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 中 1 出現的次數是 ："<<s.count(1)<<endl;
        cout<<"set 中 4 出現的次數是 ："<<s.count(4)<<endl;
        return 0;
    }

 

運行結果：

小結：count() 這個函數在set並不是很實用，因為一個鍵值在set只可能出現0或1次；

equal_range() ，返回一對定位器，分別表示第一個大於或等於給定關鍵值的元素和 第一個大於給定關鍵值的元素，這個返回值是一個pair類型，如果這一對定位器中哪個返回失敗，就會等於end()的值。具體這個有什么用途我還沒遇到過~~~

示例代碼：

  

    #include <iostream>
    #include <set>
    #include <cstdio>
    #include <algorithm>
     
    using namespace std;
     
    int main()
    {
        set<int> s;
        set<int>::iterator iter;
        for(int i = 1 ; i <= 5; ++i)
        {
            int x;
            scanf("%d",&x);
            s.insert(x); //向set中加入數據
        }
        for(iter = s.begin() ; iter != s.end() ; ++iter)
        {
            cout << *iter << " ";  //輸出set中的數據
        }
        cout<<endl;
        pair<set<int>::const_iterator,set<int>::const_iterator> pr;
        pr = s.equal_range(3);
        cout<<"第一個大於等於 3 的數是 ："<<*pr.first<<endl;
        cout<<"第一個大於 3的數是 ： "<<*pr.second<<endl;
        return 0;
    }

 

erase

erase(iterator)  ,刪除定位器iterator指向的值
erase(first,second),刪除定位器first和second之間的值
erase(key_value),刪除鍵值key_value的值

看看程序吧：

    #include <iostream>
    #include <set>
    using namespace std;
     
    int main()
    {
        set<int> s;
        set<int>::iterator it;
        set<int>::const_iterator iter;
        set<int>::iterator first;
        set<int>::iterator second;
        for(int i = 1 ; i <= 10 ; ++i)
        {
            s.insert(i);
            cout << i << "  " ;
        }
        cout << endl;
        it = s.begin();
        cout << "第一次刪除的是：" << *it << endl;
        s.erase(s.begin());
        cout << "第一種刪除之后 ："  << endl;
        for(iter = s.begin() ; iter != s.end() ; ++iter)
        {
            cout<<*iter<<" ";
        }
        cout << endl;
        first = s.begin();
        second = s.begin();
        second++;
        second++;
        cout << "l == " << *first  << "  r == " << *second << endl;
        cout << "第二種刪除之后 ："  << endl;
        s.erase(first,second);
        for(iter = s.begin() ; iter != s.end() ; ++iter)
        {
            cout<<*iter<<" ";
        }
        cout << endl;
     
        s.erase(8);
        cout << "第三次刪除的是： 8" << endl;
        cout<<"    第三種刪除后 set 中元素是 ：";
        for(iter = s.begin() ; iter != s.end() ; ++iter)
        {
            cout<<*iter<<" ";
        }
        cout << endl;
        return 0;
    }

運行結果：

小結：set中的刪除操作是不進行任何的錯誤檢查的，比如定位器的是否合法等等，所以用的時候自己一定要注意。第二種刪除是左閉右開[ l, r);
 

find

find()  ，返回給定值的定位器，如果沒找到則返回end()。

示例代碼：

   

    #include <iostream>
    #include <set>
     
    using namespace std;
     
    int main()
    {
        int a[] = { 4, 6, 2 , 3};
        set<int> s(a,a + 4);
        set<int>::iterator iter;
        for(iter = s.begin();iter != s.end();iter++)
        {
            cout << *iter << "  " ;
        }
        cout << endl;
        if((iter = s.find(2)) != s.end())
            cout<<*iter<<endl;
        if((iter = s.find(1)) != s.end())
            cout<<*iter<<endl;
        return 0;
    }

結果：

小結：find(x)是返回的是x的值，如果x沒有在set中則會輸出end()；

insert

insert(key_value); 將key_value插入到set中 ，返回值是pair<set<int>::iterator,bool>，bool標志着插入是否成功，而iterator代表插入的位置，若key_value已經在set中，則iterator表示的key_value在set中的位置。

inset(first,second);將定位器first到second之間的元素插入到set中，返回值是void.

示例代碼：

 

    #include <iostream>
    #include <set>
    using namespace std;
     
    int main()
    {
        int a[] = {1, 4, 3, 7};
        set<int> s;
        set<int>::iterator iter;
        s.insert(a,a + 4);
        for(iter = s.begin() ; iter != s.end() ; ++iter)
        {
            cout<<*iter<<" ";
        }
        cout<<endl;
        pair<set<int>::iterator,bool> pr;
        pr = s.insert(5);
        if(pr.second)
        {
            cout<<*pr.first<<endl;
        }
        cout << " 插入數值之后的set中有：" << endl;
        for(iter = s.begin() ; iter != s.end() ; ++iter)
        {
            cout<<*iter<<" ";
        }
        cout<<endl;
     
        return 0;
    }

運行結果：

x相當於一個簡單的插入操作

lower_bound

upper_bound

lower_bound(key_value) ，返回第一個大於等於key_value的定位器

upper_bound(key_value)，返回最后一個大於等於key_value的定位器

示例代碼：

    #include <iostream>
    #include <set>
     
    using namespace std;
     
    int main()
    {
        set<int> s;
        for(int i = 1;i <= 5;i ++)
        {
            int x;
            cin >> x;
            s.insert(x);
        }
        set<int>::iterator iter;
        for(iter = s.begin() ; iter != s.end() ; ++iter)
        {
            cout << *iter << " ";
        }
        cout << endl;
        cout << "第一個大於等於 2 的值是:   ";
            cout<<*s.lower_bound(2)<<endl;
        cout << "第一個大於等於 3 的值是:   ";
            cout<<*s.lower_bound(3)<<endl;
        cout << "第一個大於 3 的值是:   ";
            cout<<*s.upper_bound(3)<<endl;
        return 0;
    }

 

運行結果：


三.自定義比較函數
 

      (1)元素不是結構體：
        例：
      //自定義比較函數myComp,重載“（）”操作符
    

    struct myComp
    {
        bool operator()(const your_type &a,const your_type &b)
        {
            return a.data > b.data;
        }
    }
    set<int,myComp>s;
    set<int,myComp>::iterator it;

 

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

 

    struct Info
    {
        string name;
        float score;
        //重載“<”操作符，自定義排序規則
        bool operator < (const Info &a) const
        {
            //按score從大到小排列
            return a.score<score;
        }
    }
    set<Info> s;
    set<Info>::iterator it;