线性时间选出一个数组中第i大的数

来源:互联网 发布:荷仙姑的店淘宝假货 编辑:程序博客网 时间:2024/05/28 15:57
利用快排的partition,在线性时间内,选择出第i大的数
template<typename Comparable>int qPartition(vector<Comparable> &vec,int p,int r){uniform_int_distribution<unsigned> u(p,r);default_random_engine e(32676);int pos=u(e);std::swap(vec[pos],vec[r]);int key=vec[r];int i=p-1;for(int j=p;j<r;j++)if(vec[j]<key)std::swap(vec[++i],vec[j]);std::swap(vec[++i],vec[r]);return i;}template<typename Comparable>Comparable randomSelect(vector<Comparable> &vec,int p,int r,int i){if(p==r)return vec[p];int q=qPartition(vec,p,r);int k=q-p+1;if(i==k)return vec[k];else{if(i<k)return randomSelect(vec,p,q-1,i);elsereturn randomSelect(vec,q+1,r,i-k);} }int main(){vector<int> vec{1,9,8,5,7,12,3,5,6};//qSort(vec,0,vec.size());cout<<randomSelect(vec,0,vec.size()-1,3)<<endl;}

0 0
原创粉丝点击