PTA 7-171 第k小元素 (20分)

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

猜你喜欢

转载自blog.csdn.net/Minelois/article/details/113208755