1 // RB tree implementation -*- C++ -*-
3 // Copyright (C) 2001, 2002, 2003, 2004 Free Software Foundation, Inc.
5 // This file is part of the GNU ISO C++ Library. This library is free
6 // software; you can redistribute it and/or modify it under the
7 // terms of the GNU General Public License as published by the
8 // Free Software Foundation; either version 2, or (at your option)
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 // GNU General Public License for more details.
16 // You should have received a copy of the GNU General Public License along
17 // with this library; see the file COPYING. If not, write to the Free
18 // Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307,
21 // As a special exception, you may use this file as part of a free software
22 // library without restriction. Specifically, if other files instantiate
23 // templates or use macros or inline functions from this file, or you compile
24 // this file and link it with other files to produce an executable, this
25 // file does not by itself cause the resulting executable to be covered by
26 // the GNU General Public License. This exception does not however
27 // invalidate any other reasons why the executable file might be covered by
28 // the GNU General Public License.
32 * Copyright (c) 1996,1997
33 * Silicon Graphics Computer Systems, Inc.
35 * Permission to use, copy, modify, distribute and sell this software
36 * and its documentation for any purpose is hereby granted without fee,
37 * provided that the above copyright notice appear in all copies and
38 * that both that copyright notice and this permission notice appear
39 * in supporting documentation. Silicon Graphics makes no
40 * representations about the suitability of this software for any
41 * purpose. It is provided "as is" without express or implied warranty.
45 * Hewlett-Packard Company
47 * Permission to use, copy, modify, distribute and sell this software
48 * and its documentation for any purpose is hereby granted without fee,
49 * provided that the above copyright notice appear in all copies and
50 * that both that copyright notice and this permission notice appear
51 * in supporting documentation. Hewlett-Packard Company makes no
52 * representations about the suitability of this software for any
53 * purpose. It is provided "as is" without express or implied warranty.
59 * This is an internal header file, included by other library headers.
60 * You should not attempt to use it directly.
66 #include <bits/stl_algobase.h>
67 #include <bits/allocator.h>
68 #include <bits/stl_construct.h>
69 #include <bits/stl_function.h>
70 #include <bits/cpp_type_traits.h>
74 // Red-black tree class, designed for use in implementing STL
75 // associative containers (set, multiset, map, and multimap). The
76 // insertion and deletion algorithms are based on those in Cormen,
77 // Leiserson, and Rivest, Introduction to Algorithms (MIT Press,
80 // (1) the header cell is maintained with links not only to the root
81 // but also to the leftmost node of the tree, to enable constant
82 // time begin(), and to the rightmost node of the tree, to enable
83 // linear time performance when used with the generic set algorithms
86 // (2) when a node being deleted has two children its successor node
87 // is relinked into its place, rather than copied, so that the only
88 // iterators invalidated are those referring to the deleted node.
90 enum _Rb_tree_color
{ _S_red
= false, _S_black
= true };
92 struct _Rb_tree_node_base
94 typedef _Rb_tree_node_base
* _Base_ptr
;
95 typedef const _Rb_tree_node_base
* _Const_Base_ptr
;
97 _Rb_tree_color _M_color
;
103 _S_minimum(_Base_ptr __x
)
105 while (__x
->_M_left
!= 0) __x
= __x
->_M_left
;
109 static _Const_Base_ptr
110 _S_minimum(_Const_Base_ptr __x
)
112 while (__x
->_M_left
!= 0) __x
= __x
->_M_left
;
117 _S_maximum(_Base_ptr __x
)
119 while (__x
->_M_right
!= 0) __x
= __x
->_M_right
;
123 static _Const_Base_ptr
124 _S_maximum(_Const_Base_ptr __x
)
126 while (__x
->_M_right
!= 0) __x
= __x
->_M_right
;
131 template<typename _Val
>
132 struct _Rb_tree_node
: public _Rb_tree_node_base
134 typedef _Rb_tree_node
<_Val
>* _Link_type
;
139 _Rb_tree_increment(_Rb_tree_node_base
* __x
);
141 const _Rb_tree_node_base
*
142 _Rb_tree_increment(const _Rb_tree_node_base
* __x
);
145 _Rb_tree_decrement(_Rb_tree_node_base
* __x
);
147 const _Rb_tree_node_base
*
148 _Rb_tree_decrement(const _Rb_tree_node_base
* __x
);
150 template<typename _Tp
>
151 struct _Rb_tree_iterator
153 typedef _Tp value_type
;
154 typedef _Tp
& reference
;
155 typedef _Tp
* pointer
;
157 typedef bidirectional_iterator_tag iterator_category
;
158 typedef ptrdiff_t difference_type
;
160 typedef _Rb_tree_iterator
<_Tp
> _Self
;
161 typedef _Rb_tree_node_base::_Base_ptr _Base_ptr
;
162 typedef _Rb_tree_node
<_Tp
>* _Link_type
;
164 _Rb_tree_iterator() { }
166 _Rb_tree_iterator(_Link_type __x
)
171 { return static_cast<_Link_type
>(_M_node
)->_M_value_field
; }
175 { return &static_cast<_Link_type
>(_M_node
)->_M_value_field
; }
180 _M_node
= _Rb_tree_increment(_M_node
);
188 _M_node
= _Rb_tree_increment(_M_node
);
195 _M_node
= _Rb_tree_decrement(_M_node
);
203 _M_node
= _Rb_tree_decrement(_M_node
);
208 operator==(const _Self
& __x
) const
209 { return _M_node
== __x
._M_node
; }
212 operator!=(const _Self
& __x
) const
213 { return _M_node
!= __x
._M_node
; }
218 template<typename _Tp
>
219 struct _Rb_tree_const_iterator
221 typedef _Tp value_type
;
222 typedef const _Tp
& reference
;
223 typedef const _Tp
* pointer
;
225 typedef _Rb_tree_iterator
<_Tp
> iterator
;
227 typedef bidirectional_iterator_tag iterator_category
;
228 typedef ptrdiff_t difference_type
;
230 typedef _Rb_tree_const_iterator
<_Tp
> _Self
;
231 typedef _Rb_tree_node_base::_Const_Base_ptr _Base_ptr
;
232 typedef const _Rb_tree_node
<_Tp
>* _Link_type
;
234 _Rb_tree_const_iterator() { }
236 _Rb_tree_const_iterator(_Link_type __x
)
239 _Rb_tree_const_iterator(const iterator
& __it
)
240 : _M_node(__it
._M_node
) { }
244 { return static_cast<_Link_type
>(_M_node
)->_M_value_field
; }
248 { return &static_cast<_Link_type
>(_M_node
)->_M_value_field
; }
253 _M_node
= _Rb_tree_increment(_M_node
);
261 _M_node
= _Rb_tree_increment(_M_node
);
268 _M_node
= _Rb_tree_decrement(_M_node
);
276 _M_node
= _Rb_tree_decrement(_M_node
);
281 operator==(const _Self
& __x
) const
282 { return _M_node
== __x
._M_node
; }
285 operator!=(const _Self
& __x
) const
286 { return _M_node
!= __x
._M_node
; }
291 template<typename _Val
>
293 operator==(const _Rb_tree_iterator
<_Val
>& __x
,
294 const _Rb_tree_const_iterator
<_Val
>& __y
)
295 { return __x
._M_node
== __y
._M_node
; }
297 template<typename _Val
>
299 operator!=(const _Rb_tree_iterator
<_Val
>& __x
,
300 const _Rb_tree_const_iterator
<_Val
>& __y
)
301 { return __x
._M_node
!= __y
._M_node
; }
304 _Rb_tree_rotate_left(_Rb_tree_node_base
* const __x
,
305 _Rb_tree_node_base
*& __root
);
308 _Rb_tree_rotate_right(_Rb_tree_node_base
* const __x
,
309 _Rb_tree_node_base
*& __root
);
312 _Rb_tree_insert_and_rebalance(const bool __insert_left
,
313 _Rb_tree_node_base
* __x
,
314 _Rb_tree_node_base
* __p
,
315 _Rb_tree_node_base
& __header
);
318 _Rb_tree_rebalance_for_erase(_Rb_tree_node_base
* const __z
,
319 _Rb_tree_node_base
& __header
);
322 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
323 typename _Compare
, typename _Alloc
= allocator
<_Val
> >
326 typedef typename
_Alloc::template rebind
<_Rb_tree_node
<_Val
> >::other
330 typedef _Rb_tree_node_base
* _Base_ptr
;
331 typedef const _Rb_tree_node_base
* _Const_Base_ptr
;
332 typedef _Rb_tree_node
<_Val
> _Rb_tree_node
;
335 typedef _Key key_type
;
336 typedef _Val value_type
;
337 typedef value_type
* pointer
;
338 typedef const value_type
* const_pointer
;
339 typedef value_type
& reference
;
340 typedef const value_type
& const_reference
;
341 typedef _Rb_tree_node
* _Link_type
;
342 typedef const _Rb_tree_node
* _Const_Link_type
;
343 typedef size_t size_type
;
344 typedef ptrdiff_t difference_type
;
345 typedef _Alloc allocator_type
;
348 get_allocator() const
349 { return *static_cast<const _Node_allocator
*>(&this->_M_impl
); }
354 { return _M_impl
._Node_allocator::allocate(1); }
357 _M_put_node(_Rb_tree_node
* __p
)
358 { _M_impl
._Node_allocator::deallocate(__p
, 1); }
361 _M_create_node(const value_type
& __x
)
363 _Link_type __tmp
= _M_get_node();
365 { std::_Construct(&__tmp
->_M_value_field
, __x
); }
369 __throw_exception_again
;
375 _M_clone_node(_Const_Link_type __x
)
377 _Link_type __tmp
= _M_create_node(__x
->_M_value_field
);
378 __tmp
->_M_color
= __x
->_M_color
;
385 destroy_node(_Link_type __p
)
387 std::_Destroy(&__p
->_M_value_field
);
392 template<typename _Key_compare
,
393 bool _Is_pod_comparator
= std::__is_pod
<_Key_compare
>::_M_type
>
394 struct _Rb_tree_impl
: public _Node_allocator
396 _Key_compare _M_key_compare
;
397 _Rb_tree_node_base _M_header
;
398 size_type _M_node_count
; // Keeps track of size of tree.
400 _Rb_tree_impl(const _Node_allocator
& __a
= _Node_allocator(),
401 const _Key_compare
& __comp
= _Key_compare())
402 : _Node_allocator(__a
), _M_key_compare(__comp
), _M_node_count(0)
404 this->_M_header
._M_color
= _S_red
;
405 this->_M_header
._M_parent
= 0;
406 this->_M_header
._M_left
= &this->_M_header
;
407 this->_M_header
._M_right
= &this->_M_header
;
411 // Specialization for _Comparison types that are not capable of
412 // being base classes / super classes.
413 template<typename _Key_compare
>
414 struct _Rb_tree_impl
<_Key_compare
, true> : public _Node_allocator
416 _Key_compare _M_key_compare
;
417 _Rb_tree_node_base _M_header
;
418 size_type _M_node_count
; // Keeps track of size of tree.
420 _Rb_tree_impl(const _Node_allocator
& __a
= _Node_allocator(),
421 const _Key_compare
& __comp
= _Key_compare())
422 : _Node_allocator(__a
), _M_key_compare(__comp
), _M_node_count(0)
424 this->_M_header
._M_color
= _S_red
;
425 this->_M_header
._M_parent
= 0;
426 this->_M_header
._M_left
= &this->_M_header
;
427 this->_M_header
._M_right
= &this->_M_header
;
431 _Rb_tree_impl
<_Compare
> _M_impl
;
436 { return this->_M_impl
._M_header
._M_parent
; }
440 { return this->_M_impl
._M_header
._M_parent
; }
444 { return this->_M_impl
._M_header
._M_left
; }
448 { return this->_M_impl
._M_header
._M_left
; }
452 { return this->_M_impl
._M_header
._M_right
; }
456 { return this->_M_impl
._M_header
._M_right
; }
460 { return static_cast<_Link_type
>(this->_M_impl
._M_header
._M_parent
); }
464 { return static_cast<_Const_Link_type
>(this->_M_impl
._M_header
._M_parent
); }
468 { return static_cast<_Link_type
>(&this->_M_impl
._M_header
); }
472 { return static_cast<_Const_Link_type
>(&this->_M_impl
._M_header
); }
474 static const_reference
475 _S_value(_Const_Link_type __x
)
476 { return __x
->_M_value_field
; }
479 _S_key(_Const_Link_type __x
)
480 { return _KeyOfValue()(_S_value(__x
)); }
483 _S_left(_Base_ptr __x
)
484 { return static_cast<_Link_type
>(__x
->_M_left
); }
486 static _Const_Link_type
487 _S_left(_Const_Base_ptr __x
)
488 { return static_cast<_Const_Link_type
>(__x
->_M_left
); }
491 _S_right(_Base_ptr __x
)
492 { return static_cast<_Link_type
>(__x
->_M_right
); }
494 static _Const_Link_type
495 _S_right(_Const_Base_ptr __x
)
496 { return static_cast<_Const_Link_type
>(__x
->_M_right
); }
498 static const_reference
499 _S_value(_Const_Base_ptr __x
)
500 { return static_cast<_Const_Link_type
>(__x
)->_M_value_field
; }
503 _S_key(_Const_Base_ptr __x
)
504 { return _KeyOfValue()(_S_value(__x
)); }
507 _S_minimum(_Base_ptr __x
)
508 { return _Rb_tree_node_base::_S_minimum(__x
); }
510 static _Const_Base_ptr
511 _S_minimum(_Const_Base_ptr __x
)
512 { return _Rb_tree_node_base::_S_minimum(__x
); }
515 _S_maximum(_Base_ptr __x
)
516 { return _Rb_tree_node_base::_S_maximum(__x
); }
518 static _Const_Base_ptr
519 _S_maximum(_Const_Base_ptr __x
)
520 { return _Rb_tree_node_base::_S_maximum(__x
); }
523 typedef _Rb_tree_iterator
<value_type
> iterator
;
524 typedef _Rb_tree_const_iterator
<value_type
> const_iterator
;
526 typedef std::reverse_iterator
<iterator
> reverse_iterator
;
527 typedef std::reverse_iterator
<const_iterator
> const_reverse_iterator
;
531 _M_insert(_Base_ptr __x
, _Base_ptr __y
, const value_type
& __v
);
534 _M_copy(_Const_Link_type __x
, _Link_type __p
);
537 _M_erase(_Link_type __x
);
540 // allocation/deallocation
544 _Rb_tree(const _Compare
& __comp
)
545 : _M_impl(allocator_type(), __comp
)
548 _Rb_tree(const _Compare
& __comp
, const allocator_type
& __a
)
549 : _M_impl(__a
, __comp
)
552 _Rb_tree(const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __x
)
553 : _M_impl(__x
.get_allocator(), __x
._M_impl
._M_key_compare
)
555 if (__x
._M_root() != 0)
557 _M_root() = _M_copy(__x
._M_begin(), _M_end());
558 _M_leftmost() = _S_minimum(_M_root());
559 _M_rightmost() = _S_maximum(_M_root());
560 _M_impl
._M_node_count
= __x
._M_impl
._M_node_count
;
565 { _M_erase(_M_begin()); }
567 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>&
568 operator=(const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __x
);
573 { return _M_impl
._M_key_compare
; }
577 { return static_cast<_Link_type
>(this->_M_impl
._M_header
._M_left
); }
581 { return static_cast<_Const_Link_type
>(this->_M_impl
._M_header
._M_left
); }
585 { return static_cast<_Link_type
>(&this->_M_impl
._M_header
); }
589 { return static_cast<_Const_Link_type
>(&this->_M_impl
._M_header
); }
593 { return reverse_iterator(end()); }
595 const_reverse_iterator
597 { return const_reverse_iterator(end()); }
601 { return reverse_iterator(begin()); }
603 const_reverse_iterator
605 { return const_reverse_iterator(begin()); }
609 { return _M_impl
._M_node_count
== 0; }
613 { return _M_impl
._M_node_count
; }
617 { return size_type(-1); }
620 swap(_Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __t
);
624 insert_unique(const value_type
& __x
);
627 insert_equal(const value_type
& __x
);
630 insert_unique(iterator __position
, const value_type
& __x
);
633 insert_equal(iterator __position
, const value_type
& __x
);
635 template<typename _InputIterator
>
637 insert_unique(_InputIterator __first
, _InputIterator __last
);
639 template<typename _InputIterator
>
641 insert_equal(_InputIterator __first
, _InputIterator __last
);
644 erase(iterator __position
);
647 erase(const key_type
& __x
);
650 erase(iterator __first
, iterator __last
);
653 erase(const key_type
* __first
, const key_type
* __last
);
658 _M_erase(_M_begin());
659 _M_leftmost() = _M_end();
661 _M_rightmost() = _M_end();
662 _M_impl
._M_node_count
= 0;
667 find(const key_type
& __x
);
670 find(const key_type
& __x
) const;
673 count(const key_type
& __x
) const;
676 lower_bound(const key_type
& __x
);
679 lower_bound(const key_type
& __x
) const;
682 upper_bound(const key_type
& __x
);
685 upper_bound(const key_type
& __x
) const;
687 pair
<iterator
,iterator
>
688 equal_range(const key_type
& __x
);
690 pair
<const_iterator
, const_iterator
>
691 equal_range(const key_type
& __x
) const;
698 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
699 typename _Compare
, typename _Alloc
>
701 operator==(const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __x
,
702 const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __y
)
704 return __x
.size() == __y
.size()
705 && equal(__x
.begin(), __x
.end(), __y
.begin());
708 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
709 typename _Compare
, typename _Alloc
>
711 operator<(const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __x
,
712 const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __y
)
714 return lexicographical_compare(__x
.begin(), __x
.end(),
715 __y
.begin(), __y
.end());
718 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
719 typename _Compare
, typename _Alloc
>
721 operator!=(const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __x
,
722 const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __y
)
723 { return !(__x
== __y
); }
725 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
726 typename _Compare
, typename _Alloc
>
728 operator>(const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __x
,
729 const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __y
)
730 { return __y
< __x
; }
732 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
733 typename _Compare
, typename _Alloc
>
735 operator<=(const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __x
,
736 const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __y
)
737 { return !(__y
< __x
); }
739 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
740 typename _Compare
, typename _Alloc
>
742 operator>=(const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __x
,
743 const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __y
)
744 { return !(__x
< __y
); }
746 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
747 typename _Compare
, typename _Alloc
>
749 swap(_Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __x
,
750 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __y
)
753 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
754 typename _Compare
, typename _Alloc
>
755 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>&
756 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
757 operator=(const _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __x
)
761 // Note that _Key may be a constant type.
763 _M_impl
._M_key_compare
= __x
._M_impl
._M_key_compare
;
764 if (__x
._M_root() != 0)
766 _M_root() = _M_copy(__x
._M_begin(), _M_end());
767 _M_leftmost() = _S_minimum(_M_root());
768 _M_rightmost() = _S_maximum(_M_root());
769 _M_impl
._M_node_count
= __x
._M_impl
._M_node_count
;
775 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
776 typename _Compare
, typename _Alloc
>
777 typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::iterator
778 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
779 _M_insert(_Base_ptr __x
, _Base_ptr __p
, const _Val
& __v
)
781 _Link_type __z
= _M_create_node(__v
);
784 __insert_left
= __x
!= 0 || __p
== _M_end()
785 || _M_impl
._M_key_compare(_KeyOfValue()(__v
),
788 _Rb_tree_insert_and_rebalance(__insert_left
, __z
, __p
,
789 this->_M_impl
._M_header
);
790 ++_M_impl
._M_node_count
;
791 return iterator(__z
);
794 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
795 typename _Compare
, typename _Alloc
>
796 typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::iterator
797 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
798 insert_equal(const _Val
& __v
)
800 _Link_type __x
= _M_begin();
801 _Link_type __y
= _M_end();
805 __x
= _M_impl
._M_key_compare(_KeyOfValue()(__v
), _S_key(__x
)) ?
806 _S_left(__x
) : _S_right(__x
);
808 return _M_insert(__x
, __y
, __v
);
811 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
812 typename _Compare
, typename _Alloc
>
814 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
815 swap(_Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>& __t
)
819 if (__t
._M_root() != 0)
821 _M_root() = __t
._M_root();
822 _M_leftmost() = __t
._M_leftmost();
823 _M_rightmost() = __t
._M_rightmost();
824 _M_root()->_M_parent
= _M_end();
827 __t
._M_leftmost() = __t
._M_end();
828 __t
._M_rightmost() = __t
._M_end();
831 else if (__t
._M_root() == 0)
833 __t
._M_root() = _M_root();
834 __t
._M_leftmost() = _M_leftmost();
835 __t
._M_rightmost() = _M_rightmost();
836 __t
._M_root()->_M_parent
= __t
._M_end();
839 _M_leftmost() = _M_end();
840 _M_rightmost() = _M_end();
844 std::swap(_M_root(),__t
._M_root());
845 std::swap(_M_leftmost(),__t
._M_leftmost());
846 std::swap(_M_rightmost(),__t
._M_rightmost());
848 _M_root()->_M_parent
= _M_end();
849 __t
._M_root()->_M_parent
= __t
._M_end();
851 // No need to swap header's color as it does not change.
852 std::swap(this->_M_impl
._M_node_count
, __t
._M_impl
._M_node_count
);
853 std::swap(this->_M_impl
._M_key_compare
, __t
._M_impl
._M_key_compare
);
856 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
857 typename _Compare
, typename _Alloc
>
858 pair
<typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::iterator
,
860 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
861 insert_unique(const _Val
& __v
)
863 _Link_type __x
= _M_begin();
864 _Link_type __y
= _M_end();
869 __comp
= _M_impl
._M_key_compare(_KeyOfValue()(__v
), _S_key(__x
));
870 __x
= __comp
? _S_left(__x
) : _S_right(__x
);
872 iterator __j
= iterator(__y
);
875 return pair
<iterator
,bool>(_M_insert(__x
, __y
, __v
), true);
878 if (_M_impl
._M_key_compare(_S_key(__j
._M_node
), _KeyOfValue()(__v
)))
879 return pair
<iterator
,bool>(_M_insert(__x
, __y
, __v
), true);
880 return pair
<iterator
,bool>(__j
, false);
883 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
884 typename _Compare
, typename _Alloc
>
885 typename _Rb_tree
<_Key
, _Val
, _KeyOfValue
, _Compare
, _Alloc
>::iterator
886 _Rb_tree
<_Key
, _Val
, _KeyOfValue
, _Compare
, _Alloc
>::
887 insert_unique(iterator __position
, const _Val
& __v
)
889 if (__position
._M_node
== _M_leftmost())
893 && _M_impl
._M_key_compare(_KeyOfValue()(__v
),
894 _S_key(__position
._M_node
)))
895 return _M_insert(__position
._M_node
, __position
._M_node
, __v
);
896 // First argument just needs to be non-null.
898 return insert_unique(__v
).first
;
900 else if (__position
._M_node
== _M_end())
903 if (_M_impl
._M_key_compare(_S_key(_M_rightmost()),
905 return _M_insert(0, _M_rightmost(), __v
);
907 return insert_unique(__v
).first
;
911 iterator __before
= __position
;
913 if (_M_impl
._M_key_compare(_S_key(__before
._M_node
),
915 && _M_impl
._M_key_compare(_KeyOfValue()(__v
),
916 _S_key(__position
._M_node
)))
918 if (_S_right(__before
._M_node
) == 0)
919 return _M_insert(0, __before
._M_node
, __v
);
921 return _M_insert(__position
._M_node
, __position
._M_node
, __v
);
922 // First argument just needs to be non-null.
925 return insert_unique(__v
).first
;
929 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
930 typename _Compare
, typename _Alloc
>
931 typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::iterator
932 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
933 insert_equal(iterator __position
, const _Val
& __v
)
935 if (__position
._M_node
== _M_leftmost())
939 && !_M_impl
._M_key_compare(_S_key(__position
._M_node
),
941 return _M_insert(__position
._M_node
, __position
._M_node
, __v
);
942 // first argument just needs to be non-null
944 return insert_equal(__v
);
946 else if (__position
._M_node
== _M_end())
949 if (!_M_impl
._M_key_compare(_KeyOfValue()(__v
),
950 _S_key(_M_rightmost())))
951 return _M_insert(0, _M_rightmost(), __v
);
953 return insert_equal(__v
);
957 iterator __before
= __position
;
959 if (!_M_impl
._M_key_compare(_KeyOfValue()(__v
),
960 _S_key(__before
._M_node
))
961 && !_M_impl
._M_key_compare(_S_key(__position
._M_node
),
964 if (_S_right(__before
._M_node
) == 0)
965 return _M_insert(0, __before
._M_node
, __v
);
967 return _M_insert(__position
._M_node
, __position
._M_node
, __v
);
968 // First argument just needs to be non-null.
971 return insert_equal(__v
);
975 template<typename _Key
, typename _Val
, typename _KoV
,
976 typename _Cmp
, typename _Alloc
>
979 _Rb_tree
<_Key
,_Val
,_KoV
,_Cmp
,_Alloc
>::
980 insert_equal(_II __first
, _II __last
)
982 for ( ; __first
!= __last
; ++__first
)
983 insert_equal(*__first
);
986 template<typename _Key
, typename _Val
, typename _KoV
,
987 typename _Cmp
, typename _Alloc
>
990 _Rb_tree
<_Key
,_Val
,_KoV
,_Cmp
,_Alloc
>::
991 insert_unique(_II __first
, _II __last
)
993 for ( ; __first
!= __last
; ++__first
)
994 insert_unique(*__first
);
997 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
998 typename _Compare
, typename _Alloc
>
1000 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::erase(iterator __position
)
1003 static_cast<_Link_type
>(_Rb_tree_rebalance_for_erase(__position
._M_node
,
1004 this->_M_impl
._M_header
));
1006 --_M_impl
._M_node_count
;
1009 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
1010 typename _Compare
, typename _Alloc
>
1011 typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::size_type
1012 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::erase(const _Key
& __x
)
1014 pair
<iterator
,iterator
> __p
= equal_range(__x
);
1015 size_type __n
= std::distance(__p
.first
, __p
.second
);
1016 erase(__p
.first
, __p
.second
);
1020 template<typename _Key
, typename _Val
, typename _KoV
,
1021 typename _Compare
, typename _Alloc
>
1022 typename _Rb_tree
<_Key
, _Val
, _KoV
, _Compare
, _Alloc
>::_Link_type
1023 _Rb_tree
<_Key
,_Val
,_KoV
,_Compare
,_Alloc
>::
1024 _M_copy(_Const_Link_type __x
, _Link_type __p
)
1026 // Structural copy. __x and __p must be non-null.
1027 _Link_type __top
= _M_clone_node(__x
);
1028 __top
->_M_parent
= __p
;
1033 __top
->_M_right
= _M_copy(_S_right(__x
), __top
);
1039 _Link_type __y
= _M_clone_node(__x
);
1041 __y
->_M_parent
= __p
;
1043 __y
->_M_right
= _M_copy(_S_right(__x
), __y
);
1051 __throw_exception_again
;
1056 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
1057 typename _Compare
, typename _Alloc
>
1059 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::_M_erase(_Link_type __x
)
1061 // Erase without rebalancing.
1064 _M_erase(_S_right(__x
));
1065 _Link_type __y
= _S_left(__x
);
1071 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
1072 typename _Compare
, typename _Alloc
>
1074 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
1075 erase(iterator __first
, iterator __last
)
1077 if (__first
== begin() && __last
== end())
1080 while (__first
!= __last
) erase(__first
++);
1083 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
1084 typename _Compare
, typename _Alloc
>
1086 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
1087 erase(const _Key
* __first
, const _Key
* __last
)
1089 while (__first
!= __last
)
1093 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
1094 typename _Compare
, typename _Alloc
>
1095 typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::iterator
1096 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::find(const _Key
& __k
)
1098 _Link_type __x
= _M_begin(); // Current node.
1099 _Link_type __y
= _M_end(); // Last node which is not less than __k.
1102 if (!_M_impl
._M_key_compare(_S_key(__x
), __k
))
1103 __y
= __x
, __x
= _S_left(__x
);
1105 __x
= _S_right(__x
);
1107 iterator __j
= iterator(__y
);
1108 return (__j
== end()
1109 || _M_impl
._M_key_compare(__k
, _S_key(__j
._M_node
))) ? end() : __j
;
1112 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
1113 typename _Compare
, typename _Alloc
>
1114 typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::const_iterator
1115 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
1116 find(const _Key
& __k
) const
1118 _Const_Link_type __x
= _M_begin(); // Current node.
1119 _Const_Link_type __y
= _M_end(); // Last node which is not less than __k.
1123 if (!_M_impl
._M_key_compare(_S_key(__x
), __k
))
1124 __y
= __x
, __x
= _S_left(__x
);
1126 __x
= _S_right(__x
);
1128 const_iterator __j
= const_iterator(__y
);
1129 return (__j
== end()
1130 || _M_impl
._M_key_compare(__k
, _S_key(__j
._M_node
))) ? end() : __j
;
1133 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
1134 typename _Compare
, typename _Alloc
>
1135 typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::size_type
1136 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
1137 count(const _Key
& __k
) const
1139 pair
<const_iterator
, const_iterator
> __p
= equal_range(__k
);
1140 const size_type __n
= std::distance(__p
.first
, __p
.second
);
1144 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
1145 typename _Compare
, typename _Alloc
>
1146 typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::iterator
1147 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
1148 lower_bound(const _Key
& __k
)
1150 _Link_type __x
= _M_begin(); // Current node.
1151 _Link_type __y
= _M_end(); // Last node which is not less than __k.
1154 if (!_M_impl
._M_key_compare(_S_key(__x
), __k
))
1155 __y
= __x
, __x
= _S_left(__x
);
1157 __x
= _S_right(__x
);
1159 return iterator(__y
);
1162 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
1163 typename _Compare
, typename _Alloc
>
1164 typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::const_iterator
1165 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
1166 lower_bound(const _Key
& __k
) const
1168 _Const_Link_type __x
= _M_begin(); // Current node.
1169 _Const_Link_type __y
= _M_end(); // Last node which is not less than __k.
1172 if (!_M_impl
._M_key_compare(_S_key(__x
), __k
))
1173 __y
= __x
, __x
= _S_left(__x
);
1175 __x
= _S_right(__x
);
1177 return const_iterator(__y
);
1180 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
1181 typename _Compare
, typename _Alloc
>
1182 typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::iterator
1183 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
1184 upper_bound(const _Key
& __k
)
1186 _Link_type __x
= _M_begin(); // Current node.
1187 _Link_type __y
= _M_end(); // Last node which is greater than __k.
1190 if (_M_impl
._M_key_compare(__k
, _S_key(__x
)))
1191 __y
= __x
, __x
= _S_left(__x
);
1193 __x
= _S_right(__x
);
1195 return iterator(__y
);
1198 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
1199 typename _Compare
, typename _Alloc
>
1200 typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::const_iterator
1201 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
1202 upper_bound(const _Key
& __k
) const
1204 _Const_Link_type __x
= _M_begin(); // Current node.
1205 _Const_Link_type __y
= _M_end(); // Last node which is greater than __k.
1208 if (_M_impl
._M_key_compare(__k
, _S_key(__x
)))
1209 __y
= __x
, __x
= _S_left(__x
);
1211 __x
= _S_right(__x
);
1213 return const_iterator(__y
);
1216 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
1217 typename _Compare
, typename _Alloc
>
1219 pair
<typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,
1220 _Compare
,_Alloc
>::iterator
,
1221 typename _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::iterator
>
1222 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::
1223 equal_range(const _Key
& __k
)
1224 { return pair
<iterator
, iterator
>(lower_bound(__k
), upper_bound(__k
)); }
1226 template<typename _Key
, typename _Val
, typename _KoV
,
1227 typename _Compare
, typename _Alloc
>
1229 pair
<typename _Rb_tree
<_Key
, _Val
, _KoV
,
1230 _Compare
, _Alloc
>::const_iterator
,
1231 typename _Rb_tree
<_Key
, _Val
, _KoV
, _Compare
, _Alloc
>::const_iterator
>
1232 _Rb_tree
<_Key
, _Val
, _KoV
, _Compare
, _Alloc
>::
1233 equal_range(const _Key
& __k
) const
1234 { return pair
<const_iterator
, const_iterator
>(lower_bound(__k
),
1235 upper_bound(__k
)); }
1238 _Rb_tree_black_count(const _Rb_tree_node_base
* __node
,
1239 const _Rb_tree_node_base
* __root
);
1241 template<typename _Key
, typename _Val
, typename _KeyOfValue
,
1242 typename _Compare
, typename _Alloc
>
1244 _Rb_tree
<_Key
,_Val
,_KeyOfValue
,_Compare
,_Alloc
>::__rb_verify() const
1246 if (_M_impl
._M_node_count
== 0 || begin() == end())
1247 return _M_impl
._M_node_count
== 0 && begin() == end()
1248 && this->_M_impl
._M_header
._M_left
== _M_end()
1249 && this->_M_impl
._M_header
._M_right
== _M_end();
1251 unsigned int __len
= _Rb_tree_black_count(_M_leftmost(), _M_root());
1252 for (const_iterator __it
= begin(); __it
!= end(); ++__it
)
1254 _Const_Link_type __x
= static_cast<_Const_Link_type
>(__it
._M_node
);
1255 _Const_Link_type __L
= _S_left(__x
);
1256 _Const_Link_type __R
= _S_right(__x
);
1258 if (__x
->_M_color
== _S_red
)
1259 if ((__L
&& __L
->_M_color
== _S_red
)
1260 || (__R
&& __R
->_M_color
== _S_red
))
1263 if (__L
&& _M_impl
._M_key_compare(_S_key(__x
), _S_key(__L
)))
1265 if (__R
&& _M_impl
._M_key_compare(_S_key(__R
), _S_key(__x
)))
1268 if (!__L
&& !__R
&& _Rb_tree_black_count(__x
, _M_root()) != __len
)
1272 if (_M_leftmost() != _Rb_tree_node_base::_S_minimum(_M_root()))
1274 if (_M_rightmost() != _Rb_tree_node_base::_S_maximum(_M_root()))