BUPT-查找第K小数

题目链接

​​https://www.nowcoder.com/practice/204dfa6fcbc8478f993d23f693189ffd?tpId=67&tqId=29637&tPage=1&ru=/kaoyan/retest/1005&qru=/ta/bupt-kaoyan/question-ranking​​

题目描述

查找一个数组的第K小的数，注意同样大小算一样大。 如  2 1 3 4 5 2 第三小数为3。

输入描述:

输入有多组数据。
每组输入n，然后输入n个整数(1<=n<=1000)，再输入k。      

输出描述:

输出第k小的整数。      

示例1

输入

复制

6
2 1 3 5 2 2
3      

输出

复制

3      

#include <iostream>
#include <algorithm>
using namespace std;
bool cmp(int a, int b){
  return a > b;
}
int main(){
  int n;
  while(cin >> n){
    int k, cnt=0;;
    int buf[1000], ind[1000];
    for(int i = 0; i < 1000; i++){
      buf[i] = 0;
      ind[i] = 0;
    }
    for(int i = 0; i < n; i++){
      cin >> ind[i];        
    }
    for(int i = 0; i < n; i++){
      if(buf[ind[i]] == 0){
        buf[ind[i]] = ind[i];
        cnt++;
      }
    }
    sort(buf, buf + 1000, cmp);
    cin >> k;
    cout << buf[cnt-k] << endl;
  }
  return 0;
}