简单模拟题

题目链接：http://www.tyvj.cn/p/1001

没什么好讲的，，不过因为一个小问题一直WA，，，，素数的时候一定记得是sqrt(fabs(x))

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<string>
#include<cctype>
using namespace std;
const int maxn=10000+10;
int a[maxn];
bool cmp1(const int a,const int b)
{
    return a>b;
}
bool cmp2(const int a,const int b)
{
    return a<b;
}
bool judge(int x)
{
    bool flag=false;
    for(int i=2;i<=sqrt(fabs(x));i++)
    {
        if(x%i==0)
        {
            flag=true;
        }
    }
    return flag;
}
int main()
{
    int n,k;
    while(cin>>n>>k)
    {
        for(int i=0;i<n;i++)
            cin>>a[i];
        sort(a,a+n,cmp1);
        int k1=a[k-1];
        sort(a,a+n,cmp2);
        int k2=a[k-1];
        int m=k1-k2;
        if(judge(m))
            cout<<"NO"<<endl;
        else
            cout<<"YES"<<endl;
            cout<<m<<endl;
    }
    return 0;
}