写一个分治算法求n个数中第k大的数字。

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

// 交换两个元素的值
void swap(int *a, int *b) {
    int t = *a;
    *a = *b;
    *b = t;
}

// 快速选择算法的分区函数
int partition(int arr[], int low, int high) {
    int pivot = arr[high]; // 选择最后一个元素作为基准
    int i = (low - 1); // 较小元素的索引

    for (int j = low; j <= high - 1; j++) {
        // 如果当前元素大于或等于基准
        if (arr[j] >= pivot) {
            i++; // 增加较小元素的索引
            swap(&arr[i], &arr[j]); // 交换
        }
    }
    swap(&arr[i + 1], &arr[high]); // 将基准放到正确的位置
    return (i + 1); // 返回基准的索引
}

// 求第k大的元素
int k_th_larg(int arr[], int low, int high, int k) {
    // 找到分区点
    int pi = partition(arr, low, high);

    // 如果分区点的索引正好是k-1，则找到了第k大的元素
    if (pi == k - 1)
        return arr[pi];

    // 如果分区点的索引大于k-1，则在左半部分继续查找
    else if (pi > k - 1)
        return k_th_larg(arr, low, pi - 1, k);

    // 否则在右半部分继续查找
    else
        return k_th_larg(arr, pi + 1, high, k);
}

// 主函数
int main() {
    int n = 10, k = 4; // 从n=10个数中求第k大的数
    int arr[] = {14, 7, 11, 4, 78, 90, 2, 33, 26, 35}; // 10个数
    int result = k_th_larg(arr, 0, n - 1, k); // 求n个数中第k大数的分治算法
    printf("第%d大的元素是%d\n", k, result);
    return 0;
}