在集合中查找前k个最小的数

#include <stdio.h>
#include <stdlib.h>

#define SIZE 10

int * k_thSmallest(int * arr, int len, int k_len);

void find(int* arr, int low, int high, int k_len);

void swap(int * a, int *b);

void printArr(int * arr, int len);

int main()
{
    int arr[SIZE] = { 54,5,19,7,49,23,38,16,9,43 };
    int k_len = 4;
    int* outArr = k_thSmallest(arr, SIZE, k_len);
    printArr(outArr, k_len);
    free(outArr);
    return 0;
}

void printArr(int * arr, int len)
{
    for (int i = 0; i < len; i++)
        printf("%d ", arr[i]);
    printf("\n");
}

void swap(int * a, int *b)
{
    int temp = *a;
    *a = *b;
    *b = temp;
}

void find(int* arr, int low, int high, int k_len)
{
    int pivot = arr[high];
    int j = low;
    for (int i = low; i <= high; i++)
    {
        if (arr[i] <= pivot) {
            swap(&arr[i], &arr[j]);
            j++;
        }
    }
    j--;
    if (j == k_len)
        return;
    else if (j > k_len)
        find(arr, low, j - 1, k_len);
    else find(arr, j + 1, high, k_len);
}

int * k_thSmallest(int * arr, int len, int k_len)
{
    find(arr, 0, len - 1, k_len);
    int* result = (int*)malloc(k_len * sizeof(int));
    for (int i = 0; i < k_len; i++)
        result[i] = arr[i];
    return result;
}