leetcode40--最小的k个数

输入整数数组 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;
  }
}
发布了40 篇原创文章 · 获赞 2 · 访问量 600

猜你喜欢

转载自blog.csdn.net/scacacac/article/details/104993431
今日推荐