这道题简单的思路莫过于把输入的n个整数排序,这样排在最前面的k个数就是最小的k个数。这是这种思路的时间复杂点为O(nlogn)。
我们设最小的k个数中最大的数为A。在快速排序算法中,我们现在数组中随机选择一个数字,然后调整数组中数字的顺序,使得比选中的数字校的数字都排在它的左边,比选中的数字大的数字都在它右边(即一次快排)。如果这个选中的数字的下标刚好是k-1(下标从0开始),那么这个数字(就是A)加上左侧的k-1个数字就是最小的k个数。
注意上述方法平均复杂度为O(N),算法导论有时间复杂度为O(N)的证明。
#include<stdio.h>
int Partition(int numbers[], int start, int end)
{
int small = start - 1;
int x = numbers[end];
for (int i = start; i < end; i++)
{
if(numbers[i]< x)
{
small++;
if(small!=i){
int temp = numbers[small];
numbers[small] = numbers[i];
numbers[i] = temp;
}
}
}
++small; //分割位置
int temp1 = numbers[small];
numbers[small] = numbers[end];
numbers[end] = temp1;
return small;
}
void getLeastKNum(int numbers[],int n , int outputs[], int k)
{
int start = 0;
int end = n - 1;
int index = Partition(numbers,start , end);
while(index != k-1)
{
if(index < k-1)
{
start = index + 1;
index = Partition(numbers,start , end);
}
else
{
end = index -1;
index = Partition(numbers,start, end);
}
}
for (int s=0;s<n; s++)
{
printf("%d ",numbers[s]);
}
printf("\n");
for(int i=0; i<k;i++)
{
outputs[i] = numbers[i];
}
}
int main()
{
int numbers[] = {2 ,3,7,4,1,5};
int outputs[3];
getLeastKNum(numbers,6,outputs ,3);
for (int i = 0; i < 3; i++)
{
printf("%d\n", outputs[i]);
}
getchar();
return 1;
}