STL源碼剖析——關聯容器之multiset

前言

multiset的特性及其用法和set完全相同，唯一的差別就是multiset允許鍵值key重複，是以multiset的插入操作采用的是底層RB-Tree的insert_equal()而非insert_unique()，有關set容器的介紹前往博文《STL源碼剖析——關聯容器之set》。本文的源碼出自SGI STL中的<stl_multiset.h>檔案。

multiset容器源碼剖析

#ifndef __SGI_STL_INTERNAL_MULTISET_H
#define __SGI_STL_INTERNAL_MULTISET_H

#include <concept_checks.h>

__STL_BEGIN_NAMESPACE

#if defined(__sgi) && !defined(__GNUC__) && (_MIPS_SIM != _MIPS_SIM_ABI32)
#pragma set woff 1174
#pragma set woff 1375
#endif

//multiset的特性及其用法和set完全相同，唯一的差別就是multiset允許鍵值key重複
//是以multiset的插入操作采用的是底層RB-Tree的insert_equal()而非insert_unique()
//有關set容器的剖析見前面博文

// Forward declaration of operators < and ==, needed for friend declaration.

//multiset内部元素預設使用遞增排序less
//使用者可自行制定比較類型
//内部維護的資料結構是紅黑樹, 具有非常優秀的最壞情況的時間複雜度 
template <class _Key, class _Compare __STL_DEPENDENT_DEFAULT_TMPL(less<_Key>),
          class _Alloc = __STL_DEFAULT_ALLOCATOR(_Key) >
class multiset;

template <class _Key, class _Compare, class _Alloc>
inline bool operator==(const multiset<_Key,_Compare,_Alloc>& __x, 
                       const multiset<_Key,_Compare,_Alloc>& __y);

template <class _Key, class _Compare, class _Alloc>
inline bool operator<(const multiset<_Key,_Compare,_Alloc>& __x, 
                      const multiset<_Key,_Compare,_Alloc>& __y);

template <class _Key, class _Compare, class _Alloc>
class multiset {
  // requirements:
  
  __STL_CLASS_REQUIRES(_Key, _Assignable);
  __STL_CLASS_BINARY_FUNCTION_CHECK(_Compare, bool, _Key, _Key);

public:

  // typedefs:

  //以下的定義與set相同
  typedef _Key     key_type;
  typedef _Key     value_type;
  typedef _Compare key_compare;
  typedef _Compare value_compare;
private:
	//底層機制是采用RB-Tree
  typedef _Rb_tree<key_type, value_type, 
                  _Identity<value_type>, key_compare, _Alloc> _Rep_type;
  _Rep_type _M_t;  // red-black tree representing multiset
public:
  typedef typename _Rep_type::const_pointer pointer;
  typedef typename _Rep_type::const_pointer const_pointer;
  typedef typename _Rep_type::const_reference reference;
  typedef typename _Rep_type::const_reference const_reference;
  typedef typename _Rep_type::const_iterator iterator;
  typedef typename _Rep_type::const_iterator const_iterator;
  typedef typename _Rep_type::const_reverse_iterator reverse_iterator;
  typedef typename _Rep_type::const_reverse_iterator const_reverse_iterator;
  typedef typename _Rep_type::size_type size_type;
  typedef typename _Rep_type::difference_type difference_type;
  typedef typename _Rep_type::allocator_type allocator_type;

  // allocation/deallocation
  /*
  構造函數
		multiset();
		explicit multiset (const key_compare& comp = key_compare(),
					   const allocator_type& alloc = allocator_type());
		
		template <class InputIterator>
		multiset (InputIterator first, InputIterator last,
				const key_compare& comp = key_compare(),
				const allocator_type& alloc = allocator_type());
		
		multiset (const multiset& x);
  */

  multiset() : _M_t(_Compare(), allocator_type()) {}
  explicit multiset(const _Compare& __comp,
                    const allocator_type& __a = allocator_type())
    : _M_t(__comp, __a) {}

#ifdef __STL_MEMBER_TEMPLATES

  //multiset的插入操作采用的是底層RB-Tree的insert_equal()而非insert_unique()
  template <class _InputIterator>
  multiset(_InputIterator __first, _InputIterator __last)
    : _M_t(_Compare(), allocator_type())
    { _M_t.insert_equal(__first, __last); }

  template <class _InputIterator>
  multiset(_InputIterator __first, _InputIterator __last,
           const _Compare& __comp,
           const allocator_type& __a = allocator_type())
    : _M_t(__comp, __a) { _M_t.insert_equal(__first, __last); }

#else

  multiset(const value_type* __first, const value_type* __last)
    : _M_t(_Compare(), allocator_type())
    { _M_t.insert_equal(__first, __last); }

  multiset(const value_type* __first, const value_type* __last,
           const _Compare& __comp,
           const allocator_type& __a = allocator_type())
    : _M_t(__comp, __a) { _M_t.insert_equal(__first, __last); }

  multiset(const_iterator __first, const_iterator __last)
    : _M_t(_Compare(), allocator_type())
    { _M_t.insert_equal(__first, __last); }

  multiset(const_iterator __first, const_iterator __last,
           const _Compare& __comp,
           const allocator_type& __a = allocator_type())
    : _M_t(__comp, __a) { _M_t.insert_equal(__first, __last); }
   
#endif /* __STL_MEMBER_TEMPLATES */

  multiset(const multiset<_Key,_Compare,_Alloc>& __x) : _M_t(__x._M_t) {}
  multiset<_Key,_Compare,_Alloc>&
  operator=(const multiset<_Key,_Compare,_Alloc>& __x) {
    _M_t = __x._M_t;//調用了底層紅黑樹的operator=操作函數 
    return *this;
  }

  //以下所有的multiset操作行為，RB-tree都已提供，是以multiset隻要調用即可 
  // accessors:

  //傳回用于key比較的函數,調用RB-Tree的key_comp()
  key_compare key_comp() const { return _M_t.key_comp(); }
   //由于multiset的性質, value和key使用同一個比較函數
  value_compare value_comp() const { return _M_t.key_comp(); }
  allocator_type get_allocator() const { return _M_t.get_allocator(); }

  iterator begin() const { return _M_t.begin(); }
  iterator end() const { return _M_t.end(); }
  reverse_iterator rbegin() const { return _M_t.rbegin(); } 
  reverse_iterator rend() const { return _M_t.rend(); }
  bool empty() const { return _M_t.empty(); }
  size_type size() const { return _M_t.size(); }
  size_type max_size() const { return _M_t.max_size(); }
  //這裡調用的swap()函數是專屬于RB-Tree的swap()，并不是STL的swap()算法
  void swap(multiset<_Key,_Compare,_Alloc>& __x) { _M_t.swap(__x._M_t); }

  // insert/erase
  /*
  插入節點
	iterator insert (const value_type& val);
	
	iterator insert (iterator position, const value_type& val);
		
	template <class InputIterator>
	void insert (InputIterator first, InputIterator last);
  */
  //插入資料節點
  iterator insert(const value_type& __x) { 
    return _M_t.insert_equal(__x);
  }
  //在指定位置插入節點
  iterator insert(iterator __position, const value_type& __x) {
    typedef typename _Rep_type::iterator _Rep_iterator;
    return _M_t.insert_equal((_Rep_iterator&)__position, __x);
  }

#ifdef __STL_MEMBER_TEMPLATES  
  template <class _InputIterator>
  void insert(_InputIterator __first, _InputIterator __last) {
    _M_t.insert_equal(__first, __last);
  }
#else
  void insert(const value_type* __first, const value_type* __last) {
    _M_t.insert_equal(__first, __last);
  }
  void insert(const_iterator __first, const_iterator __last) {
    _M_t.insert_equal(__first, __last);
  }
#endif /* __STL_MEMBER_TEMPLATES */
  //擦除指定位置的元素
  void erase(iterator __position) { 
    typedef typename _Rep_type::iterator _Rep_iterator;
    _M_t.erase((_Rep_iterator&)__position); 
  }
  //擦除元素值為x的節點
  size_type erase(const key_type& __x) { 
    return _M_t.erase(__x); 
  }
  //擦除指定區間的節點
  void erase(iterator __first, iterator __last) { 
    typedef typename _Rep_type::iterator _Rep_iterator;
    _M_t.erase((_Rep_iterator&)__first, (_Rep_iterator&)__last); 
  }
  //清除multiset
  void clear() { _M_t.clear(); }

  // multiset operations:

  //查找元素值為x的節點
  iterator find(const key_type& __x) const { return _M_t.find(__x); }
  //傳回指定元素的個數
  size_type count(const key_type& __x) const { return _M_t.count(__x); }
  //Returns an iterator pointing to the first element in the container 
  //whose key is not considered to go before k (i.e., either it is equivalent or goes after).
  iterator lower_bound(const key_type& __x) const {
    return _M_t.lower_bound(__x);
  }
  //Returns an iterator pointing to the first element that is greater than key.
  iterator upper_bound(const key_type& __x) const {
    return _M_t.upper_bound(__x); 
  }
  //Returns the bounds of a range that includes all the elements in the container 
  //which have a key equivalent to k
  //Because the elements in a map container have unique keys, 
  //the range returned will contain a single element at most.
  pair<iterator,iterator> equal_range(const key_type& __x) const {
    return _M_t.equal_range(__x);
  }

  //以下是操作符的重載

#ifdef __STL_TEMPLATE_FRIENDS
  template <class _K1, class _C1, class _A1>
  friend bool operator== (const multiset<_K1,_C1,_A1>&,
                          const multiset<_K1,_C1,_A1>&);
  template <class _K1, class _C1, class _A1>
  friend bool operator< (const multiset<_K1,_C1,_A1>&,
                         const multiset<_K1,_C1,_A1>&);
#else /* __STL_TEMPLATE_FRIENDS */
  friend bool __STD_QUALIFIER
  operator== __STL_NULL_TMPL_ARGS (const multiset&, const multiset&);
  friend bool __STD_QUALIFIER
  operator< __STL_NULL_TMPL_ARGS (const multiset&, const multiset&);
#endif /* __STL_TEMPLATE_FRIENDS */
};

template <class _Key, class _Compare, class _Alloc>
inline bool operator==(const multiset<_Key,_Compare,_Alloc>& __x, 
                       const multiset<_Key,_Compare,_Alloc>& __y) {
  return __x._M_t == __y._M_t;
}

template <class _Key, class _Compare, class _Alloc>
inline bool operator<(const multiset<_Key,_Compare,_Alloc>& __x, 
                      const multiset<_Key,_Compare,_Alloc>& __y) {
  return __x._M_t < __y._M_t;
}

#ifdef __STL_FUNCTION_TMPL_PARTIAL_ORDER

template <class _Key, class _Compare, class _Alloc>
inline bool operator!=(const multiset<_Key,_Compare,_Alloc>& __x, 
                       const multiset<_Key,_Compare,_Alloc>& __y) {
  return !(__x == __y);
}

template <class _Key, class _Compare, class _Alloc>
inline bool operator>(const multiset<_Key,_Compare,_Alloc>& __x, 
                      const multiset<_Key,_Compare,_Alloc>& __y) {
  return __y < __x;
}

template <class _Key, class _Compare, class _Alloc>
inline bool operator<=(const multiset<_Key,_Compare,_Alloc>& __x, 
                       const multiset<_Key,_Compare,_Alloc>& __y) {
  return !(__y < __x);
}

template <class _Key, class _Compare, class _Alloc>
inline bool operator>=(const multiset<_Key,_Compare,_Alloc>& __x, 
                       const multiset<_Key,_Compare,_Alloc>& __y) {
  return !(__x < __y);
}

template <class _Key, class _Compare, class _Alloc>
inline void swap(multiset<_Key,_Compare,_Alloc>& __x, 
                 multiset<_Key,_Compare,_Alloc>& __y) {
  __x.swap(__y);
}

#endif /* __STL_FUNCTION_TMPL_PARTIAL_ORDER */

#if defined(__sgi) && !defined(__GNUC__) && (_MIPS_SIM != _MIPS_SIM_ABI32)
#pragma reset woff 1174
#pragma reset woff 1375
#endif

__STL_END_NAMESPACE

#endif /* __SGI_STL_INTERNAL_MULTISET_H */

// Local Variables:
// mode:C++
// End:

參考資料：

《STL源碼剖析》侯捷

STL源碼剖析——關聯容器之multiset

前言

multiset容器源碼剖析

繼續閱讀

STL源碼剖析 3、疊代器概念與traits程式設計技法疊代器Traits程式設計技法——STL源代碼門鑰iterator源代碼完整重列SGI STL的私房菜 __type_traits

STL源碼剖析 - 第3章疊代器的概念與traits程式設計技法3.4 Traits程式設計技術3.5 疊代器3.6 疊代器iterator源碼3.7 SGI STL的__type_traits技術總結

STL源碼剖析疊代器（iterator）概念與程式設計技法（三）

STL中各個容器的實作基本原理以及互相依賴

stl源碼剖析_C++ STL源碼剖析之容器配接器stack與queue、priority_queue

STL源碼剖析——序列容器之list

stl源碼剖析（2）空間配置器一個簡單的allocator配置器構造和析構基本工具：construct()和destroy()空間的配置設定與釋放：std::alloc

C++實作一個簡單的紅黑樹（RB_TREE）

STL 源碼剖析（四）序列式容器--vector1. 寫在前面2. vector3.vector元素操作：pop_back,erase,clear,insert

STL源碼分析之vector容器

STL源碼剖析——關聯容器之multimap

STL源碼剖析——hash_map和hash_multimap

STL源碼——序列式容器deque

STL源碼筆記之序列式容器

STL源碼剖析（一）寫在前面1.了解STL2. 組态設定

STL 源碼剖析算法 stl_algo.h -- nth_element