质数

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