【Tyvj】第k极值

描述 Description  给定一个长度为N(0<n<=10000)的序列，保证每一个序列中的数字a[i]是小于maxlongint的非负整数 ，编程要求求出整个序列中第k大的数字减去第k小的数字的值m，并判断m是否为质数。(0<k<=n)       输入格式 Input Format  输入格式：
第一行为2个数n，k（含义如上题）
第二行为n个数，表示这个序列       输出格式 Output Format  输出格式：
如果m为质数则
第一行为'YES'(没有引号）
第二行为这个数m
否则 
第一行为'NO'
第二行为这个数m

Sample Input

5 2
1 2 3 4 5

Sample Output

YES
2

时间限制 Time Limitation  各个测试点1s
数据范围
20%数据满足0<n<=10
50%数据满足0<n<=5000
100%数据满足0<n<=10000

a[i]<=maxlongint       注释 Hint  对于第K大的详细解释:
如果一个序列为1 2 2 2 2 3
第1大 为3
第2大 为2
第3大 为2
第4大 为2
第5大 为1
第K小与上例相反

另外需要注意的是
最小的质数是2,如果小于2的话,请直接输出NO最需要注意的是，最小的质数是2，如果小于2的话，直接输出NO

#include<iostream>#include<cstdio>#include<cmath>#include<algorithm>#include<cstdlib>using namespace std;int  n,k,s[10000];int judge(int x){    if (x<2) return 2;    if (x==2) return 5;    for (int i=2;i<=sqrt(x);++i)    {        if (x%i==0)        {              return  2;        }    }    return 5;}  int main(){    freopen("第K极值.in","r",stdin);    freopen("第K极值.out","w",stdout);    scanf("%d %d\n",&n,&k);    for (int i=1;i<=n;++i)    {        scanf("%d",&s[i]);     }    sort(s+1,s+n+1);    int ans=s[n-k+1]-s[k];    if (judge(ans)==5)    {cout<<"YES"<<endl;}    else cout<<"NO"<<endl;    printf("%d",ans);    return 0;}