输入整数数组 arr ,找出其中最小的 k 个数。例如,输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。
示例一:
输入:arr = [3,2,1], k = 2
输出:[1,2] 或者 [2,1]
示例二:
输入:arr = [0,1,2,1], k = 1
输出:[0]
解题思路:
1、排序(但是对排序方法有要求,冒泡排序时间复杂度太高,不能通过)
2、选择前k个元素输出、
代码:
//快速排序
void Fastsort(int* arr, int low, int high) {
if (low >= high) {
return;
}
int temp = arr[low];
int first = low;
int last = high;
while (first < last) {
while (arr[last]>temp&&last >= first) {
last--;
}
arr[first] = arr[last];
while (arr[first] <= temp&&last>first) {
first++;
}
arr[last] = arr[first];
}
arr[first] = temp;
Fastsort(arr, low, first - 1);
Fastsort(arr, first + 1, high);}
int* getLeastNumbers(int* arr, int arrSize, int k, int* returnSize){
if(arr==NULL)
return NULL;
*returnSize=k;
int tem;
int* str=(int*)malloc(k*sizeof(int));
Fastsort(arr,0,arrSize-1);
for(int i=0;i<k;i++) {
str[i]=arr[i];
}
return str;
}
}