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