输入 n 个整数的一维数组,输出其中最小的 k 个
在面试时问到这个问题时,要注意时间复杂度,空间复杂度。
大神JULY在http://blog.csdn.net/v_JULY_v/article/details/6370650中详细讲解了多种方法,我这里只写几种简单点的方法。
1、利用STL库中的partial_sort函数,对原数组 部分排序
函数原型partial_sort(beg,mid,end,comp)
//利用STL中的partial函数
void Search(int *a,int n,int k)
{
int *b=new int[n];
for(int i=0;i<n;i++)
b[i]=a[i];
partial_sort(b,b+k,b+n,less<int>());
for(int i=0;i<k;i++)
cout<<b[i]<<endl;
delete b;
}
2、利用快速排序对数组排序
利用快速排序对数组排序,时间复杂度为O(n*log2n),空间复杂度为O(log2n)
因为是用的递归方法,所以数据不能太大,否则会栈崩溃。
void QuickSort(int a[],int first,int last)
{
if(first<last)
{
int i=first,j=last;
while(i!=j)
{
if(i<j)//抓大放小
{
if(a[i]>a[j])
{
swap(a[i],a[j]);
swap(i,j);
j++;
}
else
{
j--;
}
}
if(i>j)//抓小放大
{
if(a[i]<a[j])
{
swap(a[i],a[j]);
swap(i,j);
j--;
}
else
{
j++;
}
}
}
QuickSort(a,first,i-1);
QuickSort(a,i+1,last);
}
}
3、选择排序
每次选择最小值,只要k次即可,后面的无需再排。时间复杂度为O(n*k),空间复杂度为O(1)
<span style="font-size:18px;">void SelectSort(int a[],int n,int k)
{
for(int i=0;i<k;i++)//只需要排k次
{
int min=a[i],pos=i;
//从i+1到n之间找最小值
for(int j=i+1;j<n;j++)
{
if(min>a[j])
{
min=a[j];
pos=j;
}
}
a[pos]=a[i];
a[i]=min;
}
}</span>
4、利用堆排序,建立n个元素的小顶堆
利用堆排序,建立小顶堆,每次取堆首即最小值,取k次即可。时间复杂度为O(k*log2