例如輸入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);
}