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