牛客网 查找第K小数

题目链接：https://www.nowcoder.com/practice/204dfa6fcbc8478f993d23f693189ffd?tpId=40&tqId=21522&tPage=2&rp=2&ru=/ta/kaoyan&qru=/ta/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>
#include <string>
#include <cstring>
#include <stack>
#include <cstdio>
#include <queue>
using namespace std;
int n,m;
int a[1005];
int main()
{
    while(cin>>n){
        for(int i=0;i<n;i++) cin>>a[i];
        cin>>m;
        sort(a,a+n);
        for(int i=1;i<n;i++){
            if(a[i]>a[i-1]){
                m--;
                if(m==1){
                    cout<<a[i]<<endl;
                    break;
                }
            }
        }
    }
    return 0;
}