/*
*寻找最小的k个数
*题目描述:5.查找最小的k个元素
*题目:输入n个整数,输出其中最小的k个。
*例如输入1,2,3,4,5,6,7和8这8个数字,则最小的4个数字为1,2,3和4。
*
* 先快速排序,然后输出前k个元素
* 时间复杂度O(n*logn)
*/
#include <iostream>
#include <vector>
using namespace std;
void exchange(vector<int>& vi,int i,int j)
{
int tmp = vi[i];
vi[i] = vi[j];
vi[j] = tmp;
}
int partition(vector<int>& vi,int start,int end)
{
int key = vi[end];
int i = start -1;
for(int j = start;j < end;++j)
{
if(vi[j] < key)
{
++i;
exchange(vi,i,j);
}
}
exchange(vi,i+1,end);
//cout<<i+1<<endl;
return i+1;
}
void quicksort(vector<int>& vi,int start,int end)
{
if(start < end)
{
int q = partition(vi,start,end);
//cout<<q<<endl;
quicksort(vi,start,q-1);
quicksort(vi,q+1,end);
}
}
int main(int argc,char** argv)
{
vector<int> vi;
int k = 3;
vi.push_back(1);
vi.push_back(5);
vi.push_back(4);
vi.push_back(6);
vi.push_back(2);
vi.push_back(3);
vi.push_back(8);
vi.push_back(7);
//cout<<vi.size()<<endl;
quicksort(vi,0,vi.size()-1);
for(int i = 0;i < k;++i)
{
cout<<vi[i]<<" ";
}
cout<<endl;
return 0;
}
寻找最小的k个数(先快速排序,然后输出前k个元素)
最新推荐文章于 2019-05-06 11:25:33 发布