std::sort的底层实现

最新推荐文章于 2024-09-14 13:41:32 发布

DogDu

最新推荐文章于 2024-09-14 13:41:32 发布

阅读量512

点赞数 7

文章标签：算法 c++

本文链接：https://blog.csdn.net/qq_16762209/article/details/135584435

版权

本文解析了C++标准库中sort函数背后的实现原理，介绍了__introsort_loop的递归策略，它结合了快速排序（quick_sort）、堆排序(heap_sort)和插入排序(insert_sort)，通过深度限制优化性能。

摘要由CSDN通过智能技术生成

  template<typename _RandomAccessIterator, typename _Compare>
    _GLIBCXX20_CONSTEXPR
    inline void
    __partial_sort(_RandomAccessIterator __first,
		   _RandomAccessIterator __middle,
		   _RandomAccessIterator __last,
		   _Compare __comp)
    {
      std::__heap_select(__first, __middle, __last, __comp);
      std::__sort_heap(__first, __middle, __comp);
    }  

/// This is a helper function for the sort routine.
  template<typename _RandomAccessIterator, typename _Size, typename _Compare>
    _GLIBCXX20_CONSTEXPR
    void
    __introsort_loop(_RandomAccessIterator __first,
		     _RandomAccessIterator __last,
		     _Size __depth_limit, _Compare __comp)
    {
      while (__last - __first > int(_S_threshold))
	{
	  if (__depth_limit == 0)
	    {
	      std::__partial_sort(__first, __last, __last, __comp);
	      return;
	    }
	  --__depth_limit;
	  _RandomAccessIterator __cut =
	    std::__unguarded_partition_pivot(__first, __last, __comp);
	  std::__introsort_loop(__cut, __last, __depth_limit, __comp);
	  __last = __cut;
	}
    }  

template<typename _RandomAccessIterator, typename _Compare>
    _GLIBCXX20_CONSTEXPR
    inline void
    __sort(_RandomAccessIterator __first, _RandomAccessIterator __last,
	   _Compare __comp)
    {
      if (__first != __last)
	{
	  std::__introsort_loop(__first, __last,
				std::__lg(__last - __first) * 2,
				__comp);
	  std::__final_insertion_sort(__first, __last, __comp);
	}
    }

通过上面的源码，我们知道了sort实际上使用了三种排序方法：heap_sort,quick_sort,insert_sort。观察快排的循环，会发现快排使用了while循环降低了递归深度，可以进行一次模拟，这个和红黑树中的 M_erase,M_copy函数不谋而合（可以看我的红黑树课设。）而且，函数对快排进行了深度的限制，即：__lg(last-first)*2，一旦超过这个深度，那么快排停止，如果剩余元素大于15则使用heap_sort进行排序，不然，则使用insert_sort进行排序。