选择问题（线性时间复杂度）

来源：互联网发布：命令行启动mysql服务编辑：程序博客网时间：2024/06/09 15:10

采用分治策略找出第K小的元素！要求程序的时间复杂度为线性函数。

#include<iostream>#include<iterator>#include<algorithm>#include<time.h>#include<vector>using namespace std;/**选择问题（线性时间复杂度）*在beg和end之间查找第k个元素*/int Fast_find(vector<int> &vec,int beg,int end,int k){if(k>end||k<beg+1)throw range_error("输入的参数错误");if(beg+1>=end){return vec[beg];}int i ,j;for (i = beg,j = end; i < j; ){do{j--;} while (vec[j]>vec[beg]);do{i++;}while(i<end&&vec[i]<vec[beg]);if(i<j)swap(vec[i],vec[j]);}swap(vec[beg],vec[j]);if(j==k-1) return vec[j];if(j>k-1)  return Fast_find(vec,beg,j,k);if(j<k-1)  return Fast_find(vec,j+1,end,k);}int main(){vector<int> vec;copy(istream_iterator<int>(cin),istream_iterator<int>(),back_inserter(vec));cin.clear();int k ;cout<<"输入查找第几个元素（从小到大）！"<<endl;cin>>k;long start, end;int res;start = clock();try{ res = Fast_find(vec,0,vec.size(),k);}catch(range_error ex){cout<<ex.what()<<endl;return 0;}end = clock();cout<<"第"<<k<<"个元素"<<res<<endl;cout<<"验证！"<<endl;sort(vec.begin(),vec.end());cout<<"第"<<k<<"个元素"<<vec[k-1]<<endl;cout <<"程序运行时间（单位：毫秒）： "<< end-start <<endl;return 0;}

1 0