质数

1117: 质数

时间限制: 1 Sec  内存限制: 128 MB
提交: 6  解决: 3
[提交][状态][讨论版][命题人:外部导入]

题目描述

        给定一个长度为N(0< n< =10000)的序列，保证每一个序列中的数字a[i]是小于maxlongint的非负整数  ，编程要求求出整个序列中第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

 

解题思路：sort排序，然后判断差是否为质数，值得注意的是，负数是没有质数的，之前没考虑到这个，wa了两次，有毒

#include<bits/stdc++.h>
using namespace std;

int main()
{
    int n,k;
    while(cin>>n>>k)
    {
        int a[10001];
        bool flag=false;
        for(int i=0;i<n;++i)
        {
            cin>>a[i];
        }
        sort(a,a+n);
        int sa=a[n-k]-a[k-1];
        int j=sqrt(sa);
        if(sa==2){cout<<"YES"<<endl<<sa<<endl;flag=true;}
        else if(sa<2){cout<<"NO"<<endl<<sa<<endl;flag=true;}
        else for(int i=3;i<=j;++i)
        {
            if(sa%i==0){cout<<"NO"<<endl<<sa<<endl;flag=true;break;}
        }
        if(!flag)cout<<"YES"<<endl<<sa<<endl;
    }
    return 0;
}