7-171 第k小元素 (20分)
给定一个大小为n(1≤n≤1000000)且无序的整型数组,数组中可能存在相同元素,请找出该数组第k(1≤k≤n)小的元素,第k小元素指的是按从小到大排序后的第k个位置上的元素。
输入格式:
每个输入文件为一个测试用例,每个文件的第一行给出两个正整数n和k,第二行给出n个整数,其间以空格分隔。
输出格式:
输出第k小元素的值。
输入样例:
10 4
2 3 5 12 4 9 3 8 2 9
输出样例:
3
#include<bits/stdc++.h>
using namespace std;
vector<int> num;
int main(){
int n,k,x;
cin>>n>>k;
while(n--)cin>>x,num.push_back(x);
nth_element(num.begin(),num.begin()+k-1,num.end());
cout<<num[k-1];
}