首页 > 代码库 > 快速选择(quick_select) 算法分析
快速选择(quick_select) 算法分析
快速选择算法,就是从给定的一个集合S={a1,a2,...an}中选出第K个大小的数,或者给出其所在的下标之类的。
如果使用排序,比如merge_sort,然后返回第K个元素的下标,复杂度是O(NlogN)
如果使用heap_sort,或者优先队列,则复杂度是O(NlogK)
如果使用quick _sort的一个变种,叫 quick select,则平均复杂度为O(N),最坏复杂度为O(N^2)
如果使用排序,比如merge_sort,然后返回第K个元素的下标,复杂度是O(NlogN)
如果使用heap_sort,或者优先队列,则复杂度是O(NlogK)
如果使用quick _sort的一个变种,叫 quick select,则平均复杂度为O(N),最坏复杂度为O(N^2)
如果使用一种线性选择算法,则可以达到最坏O(N)的复杂度,不过实际应用中,该算法通常比quick select慢1到2倍,所以并不常用(参考Blum, Floyd, Pratt, Rivest, and Tarjan 1973 Time bounds for selection)
算法思想:
(1)利用快速排序的分治思想,求得待搜索数组按照的主元S[q](pivot)(主元的选定有好几种方法,这里不详细讨论,可参考快速排序),以主元为界分成左右两个区间
(2)通过比较主元的位置,判断第K个大小的数在主元左区间?在主元又区间?还是就是主元?(还要注意边界条件的判断,有可能在边界)
(3)进入子区间递归调用
这里实现了stl风格的quick select,仅仅作为一个mark
#include <algorithm> #include <cassert> namespace algorithm { template<typename _Tp> const _Tp& choose_pivot(const _Tp& x, const _Tp& y, const _Tp& z) { if( (x < y && y < z)||(z < y && y < x) ) return y; else if( (z < x && x < y)||(y < x && x < z) ) return x; else return z; } template<typename _Tp,typename _Compare> const _Tp& choose_pivot(const _Tp& x, const _Tp& y,const _Tp& z, _Compare comp) { if( (comp(x,y) && comp(y,z))||(comp(z,y)&&comp(y,x)) ) return y; else if( (comp(z,x) && comp(x,y))||(comp(y,x)&&comp(x,z))) return x; return z; } template<typename _RandomAccessIterator,typename _Tp> _RandomAccessIterator quick_partition(_RandomAccessIterator first, _RandomAccessIterator last,_Tp pivot) { while( true ){ while( *first < pivot ) ++first; --last; while( pivot < *last ) --last; if( first >= last ) return first; std::swap(*first,*last); ++first; } } template<typename _RandomAccessIterator,typename _Tp, typename _Compare> _RandomAccessIterator quick_partition(_RandomAccessIterator first, _RandomAccessIterator last, _Tp pivot, _Compare comp) { while( true ){ while( comp(*first,pivot) == true ) ++first; --last; while( comp(pivot,*last) == true ) --last; if( first >= last ) return first; std::swap(*first,*last); ++first; } } template<typename _RandomAccessIterator> _RandomAccessIterator quick_select(_RandomAccessIterator first, _RandomAccessIterator last, size_t kth) { typedef typename std::iterator_traits<_RandomAccessIterator>::value_type _ValueType; typedef typename std::iterator_traits<_RandomAccessIterator>::difference_type _DistanceType; if( first == last || last-first <=(_DistanceType)kth )//out of range return last; _ValueType pivot; _RandomAccessIterator mid; while( true ) { if( kth == 0 ) return std::min_element(first,last); else if( first+kth == last - 1 ) return std::max_element(first,last); else{ mid = first+(last-first)/2; pivot = choose_pivot(*first,*mid,*(last-1)); mid = quick_partition(first,last,pivot); if( mid-first > (_DistanceType)kth ) last = mid; else{ kth -= mid-first; first = mid; } } assert( last-first > (_DistanceType)kth); } } template<typename _RandomAccessIterator,typename _Compare> _RandomAccessIterator quick_select(_RandomAccessIterator first, _RandomAccessIterator last, size_t kth,_Compare comp) { typedef typename std::iterator_traits<_RandomAccessIterator>::value_type _ValueType; typedef typename std::iterator_traits<_RandomAccessIterator>::difference_type _DistanceType; if( first == last || last-first <=(_DistanceType)kth )//out of range return last; _ValueType pivot; _RandomAccessIterator mid; while( true ) { if( kth == 0 ) return std::min_element(first,last,comp); else if( first+kth == last - 1 ) return std::max_element(first,last,comp); else{ mid = first+(last-first)/2; pivot = choose_pivot(*first,*mid,*(last-1),comp); mid = quick_partition(first,last,pivot,comp); if( mid-first > (_DistanceType)kth ) last = mid; else{ kth -= mid-first; first = mid; } } assert( last-first > (_DistanceType)kth); } } } //namespace
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。