JZ40 最小的K个数 C语言

这个就是一个典型的归并排序，就是从中点开始将数组分为两个数组，然后先将他们分别进行从小到大的排序，然后一个数组一个指针依次向后便利，比较谁小，小的放入提前准备好的空数组里，最后将数组赋给原数组，实现归并排序，然后就是用for循环将前k个数输出，就OK啦~

描述

给定一个长度为 n 的可能有重复值的数组，找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4(任意顺序皆可)。

数据范围：0≤k,n≤10000，数组中每个数的大小0≤val≤1000

要求：空间复杂度 O(n) ，时间复杂度 O(nlogk)

示例1

输入：

[4,5,1,6,2,7,3,8],4 

复制返回值：

[1,2,3,4]

复制说明：

返回最小的4个数即可，返回[1,3,2,4]也可以        

示例2

输入：

[1],0

复制返回值：

[]

复制

示例3

输入：

[0,1,2,1,2],3

复制返回值：

[0,1,1]

代码：

void merge_sort(int q[],int l,int r)
{
    if(l>=r) return;
    int mid = (l + r) / 2;
    merge_sort(q,l,mid);
    merge_sort(q,mid+1,r);
    int i=l,j=mid+1,cnt=0;
    int tmp[r-l+1];
    while(i<=mid && j<=r)
    {
        if(q[i]<q[j]) tmp[cnt++]=q[i++];
        else tmp[cnt++]=q[j++];
    }
    while(i<=mid) tmp[cnt++]=q[i++];
    while(j<=r) tmp[cnt++]=q[j++];
    for(int i=l,j=0;i<=r;i++,j++)
    {
        q[i]=tmp[j];
    }
}

int* GetLeastNumbers_Solution(int* input, int inputLen, int k, int* returnSize ) {
    int* result = (int*)malloc(sizeof(int) * k);
    *returnSize = k;
    
    merge_sort(input,0,inputLen-1);
    for(int i=0;i<k;i++)
    {
        result[i] = input[i];
    }
    
    return result;
}