【WZOI】（基础题库）第476题 第K极值

题目描述:

给定一个长度为N(0< n< =10000)的序列，保证每一个序列中的数字a[i]是小于int的非负整数  ，编程要求求出整个序列中第k大的数字减去第k小的数字的值m，并判断m是否为质数。(0< k< =n)

输入格式:

输入格式： 第一行为2个数n，k（含义如上题） 第二行为n个数，表示这个序列

输出格式:

输出格式： 如果m为质数则 第一行为'YES'(没有引号） 第二行为这个数m 否则  第一行为'NO' 第二行为这个数m

样例输入:

5 2
1 2 3 4 5

样例输出:

YES
2

提示:

对于第K大的详细解释: 如果一个序列为1  2  2  2  2  3  第1大 为3 ，第2大为2，第3大为2，第4大为2，第5大为1，第K小与上例相反 另外需要注意的是最小的质数是2,如果小于2的话或为负数时,请直接输出NO

时间限制: 1000ms
空间限制: 256MB

练习sort函数的使用的题目，简单

代码实现：

#include<bits/stdc++.h>
using namespace std;
int n,a[100005],m;
bool f(int x){//判断质数
	if(x<2){
		return 0;
	}
    if(x==2){
        return 1;
    }
    for(int i=2;i*i<=x;i++){
        if(x%i==0){
			return 0;
        }
    }
    return 1;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
         cin>>a[i];
    }
    sort(a,a+n+1);
    if(f(a[n-m+1]-a[m])){
       cout<<"YES";
    }
    else{
       cout<<"NO";
    }
    cout<<"\n"<<a[n-m+1]-a[m];
    return 0;
}