厦大C语言上机 1486 第K高的人

1486.第K高的人


时间限制: 1000 MS          内存限制: 65536 K
        
提交数: 597 (1 users)          通过数: 288 (283 users)


问题描述


对于男生来说，身高是一个比较“严肃”的话题：谁都希望自己能长高一点。很不幸，小明连170cm的门槛都没能迈进。也不知道小明从哪弄来一份全院男生的资料，不过小明最关心的还是身高！现在小明想从全院n个男生中找出第k高的男生的身高是多少，不过由于信息学院人口众多，小明想尽可能用快的方法找到他想知道的第k高的身高！


输入格式


第一行输入总男生人数n和需查询的第k高的身高（0<k<=n<100,50<n<100）
第二行输入n个身高值[150,250]，保证n个身高值互不相同。


输出格式
输出第k高的身高值。


样例输入
10 6
160 175 180 165 170 171 169 185 172 168


样例输出
170


来源

xmu

#include <stdio.h>

int partition(int *a, int low, int high)
{
    int mid = (low + high) / 2;
    a[0] = a[mid];
    a[mid] = a[low];

    while (low < high)
    {
        while (low < high && a[high] <= a[0])
            --high;
        a[low] = a[high];
        while (low < high && a[low] >= a[0])
            ++low;
        a[high] = a[low];
    }
    a[low] = a[0];

    return low;
}

int k_most(int *a, int low, int high, int k)
{
    int pivot = partition(a, low, high);
    if (pivot == k)
        return a[pivot];
    else if (pivot < k)
        return k_most(a, pivot+1, high, k);
    else
        return k_most(a, low, pivot-1, k);
}

int main()
{
    int n, k;
    int array[105] = { 0 };
    int i;

    scanf("%d %d", &n, &k);
    for (i = 1; i <= n; ++i)
        scanf("%d", &array[i]);
    printf("%d\n", k_most(array, 1, n, k));

    return 0;
}