厦大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;
}