因该方法实现与之前记录的不同,就想着记录下。。。
重点inplace_partition
函数的实现
注意:迭代器范围都是左闭右开的区间
std::iter_swap(last_greater, --first_equal);
if(comp(*last_greater, pivot))
{
std::iter_swap(last_greater++, last_less++);
}
else if(comp(pivot, *last_greater))
{
++last_greater;
}
std::swap_ranges(first_equal, last, last_less);
first_equal = last_less;
swap_ranges()函数原型如下:
template<class ForwardIterator1, class ForwardIterator2>
ForwardIterator2 swap_ranges (ForwardIterator1 first1, ForwardIterator1 last1,
ForwardIterator2 first2)
{
while (first1!=last1) {
swap (*first1, *first2);
++first1; ++first2;
}
return first2;
}
快速排序代码:
#ifndef SORTS_QUICK_SORT_H_
#define SORTS_QUICK_SORT_H_
#include <functional>
#include <iterator>
#include <algorithm>
#include <utility>
namespace detail{
template <typename T, typename Compare = std::less<T>>
const T& median(const T& a, const &b, const &c, Compare comp = Compare())
{
if(comp(a, b) and comp(b, c) or comp(c, b) and comp(b, a))
return b;
else if(comp(b,c) and comp(c, a) or comp(a,c) and (c, b))
return c;
else
return a;
}
template <typename Iter,
typename T = typename std::iterator_traits<Iter>::value_type,
typename Compare = std::less<T>>
const T& iter_median(Iter a, Iter b, Iter c, Compare comp = Compare())
{
return median(*a, *b, *c, comp);
}
template <typename BidirIt,
typename T = typename std::iterator_traits<BidirIt>::value_type,
typename Compare = std::less<T>>
std::pair<BidirIt, BidirIt> inplace_partition(BidirIt first,
BidirIt last,
const T& pivot,
Compare comp = Compare())
{
BidirIt last_less, last_greater, first_equal, last_equal;
for(last_less = first, last_greater = first, first_equal = last;
last_greater != first_equal; )
{
if(comp(*last_greater, pivot))
{
std::iter_swap(last_greater++, last_less++);
}
else if(comp(pivot, *last_greater))
{
++last_greater;
}
else
{
std::iter_swap(last_greater, --first_equal);
}
}
const auto cnt = std::distance(first_equal, last);
std::swap_ranges(first_equal, last, last_less);
first_equal = last_less;
last_equal = first_equal+cnt;
return {first_equal, last_equal};
}
}
template <typename BidirIt,
typename T = typename std::iterator_traits<BidirIt>::value_type,
typename Compare = std::less<T>>
void quick_sort(BidirIt first, BidirIt last, Compare comp = Compare())
{
for(auto size = std::distance(first, last); size > 1; size = std::distance(first, last))
{
const T pivot = detail::iter_median(first, last-1, first + size / 2, comp);
const auto eq = detail::inplace_partition(first, last, pivot, comp);
quick_sort(first, eq.first, comp);
first = eq.second;
}
}
#endif