例如输入1,2,3,4,5,6,7和8这8个数字,则最小的4个数字为1,2,3和4。
分析:这道题最简单的思路莫过于把输入的n个整数排序,这样排在最前面的k个数就是最小的k个数。只是这种思路的时间复杂度为O(nlogn)。我们试着寻找更快的解决思路。
我们可以开辟一个长度为k的数组。每次从输入的n个整数中读入一个数。如果数组中已经插入的元素少于k个,则将读入的整数直接放到数组中。否则长度为k的数组已经满了,不能再往数组里插入元素,只能替换了。如果读入的这个整数比数组中已有k个整数的最大值要小,则用读入的这个整数替换这个最大值;如果读入的整数比数组中已有k个整数的最大值还要大,则读入的这个整数不可能是最小的k个整数之一,抛弃这个整数。这种思路相当于只要排序k个整数,因此时间复杂可以降到O(n+nlogk)。通常情况下k要远小于n,所以这种办法要优于前面的思路。
#include<iostream>
using namespace std;
void quicksort(int *a,int left,int right)
{
int i=left;
int j=right;
int p=a[i];
int temp;
while(i<j)
{
while(i<j&&a[j]<=p)
j--;
if(i<j)
{
temp=a[j];
a[j]=p;
a[i]=temp;
i++;
}
while(i<j&&a[i]>=p)
i++;
if(i<j)
{
temp=a[i];
a[i]=p;
a[j]=temp;
j--;
}
}
if(i!=left)
quicksort(a,left,i-1);
//if(j!=right)
//quicksort(a,j+1,right);
}
void output(int *a,int length)
{
for(int i=0;i<length;i++)
cout<<a[i]<<" ";
}
void Mink(int *a,int length,int k)
{
int *ret=new int[k];
for(int i=0;i<k;i++)
ret[i]=a[i];
//quicksort(ret,0,k-1);
for(int i=k;i<length;i++)
{
quicksort(ret,0,k-1);
if(a[i]<ret[0])
ret[0]=a[i];
}
output(ret,k);
}
int main()
{
//int a[6]={9,4,2,7,5,1};
//quicksort(a,0,5);
//output(a,6);
int a[10]={1,2,3,4,5,6,7,8,9,10};
Mink(a,10,4);
}