找数 题解

找数（number.cpp）
【问题描述】
给定一个长度为N(0<n<=10000)的序列，保证每一个序列中的数字a[i]是小于long long int的非负整数 ，编程要求求出整个序列中第k大的数字减去第k小的数字的值m，并判断m是否为质数。(0<k<=n)。
【输入格式】
第一行为2个数n，k（含义如上题）；
第二行为n个数，表示这个序列。
【输出格式】
如果m为质数则：第一行为’YES’(没有引号），第二行为这个数m；否则第一行为’NO’
第二行为这个数m。
【时限，空间要求】
1秒，128M。
【输入输出样例】
【数据规模与约定】
20%数据满足0<n<=10
50%数据满足0<n<=5000
100%数据满足0<n<=10000
【提示】
对于第K大的详细解释:
如果一个序列为1 2 2 2 3
第1大 为3
第2大 为2
第3大 为2
第4大 为2
第5大 为1
第K小与上例相反
另外需要注意的是：最小的质数是2,如果小于2的话,请直接输出NO。
number.in
number.out
5 2
1 2 3 4 5
YES
2

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath> 
#define FILE_IO
using namespace std;
long long n,k,numbers[10005],m;

int zhishu(long long x);

int main(void)
{
	#ifdef FILE_IO
		freopen("number.in","r",stdin);
		freopen("number.out","w",stdout);
	#endif
	scanf("%I64d %I64d",&n,&k);
	for(int i=1;i<=n;i++)
		scanf("%I64d",&numbers[i]);
	sort(numbers+1,numbers+n+1);
	m=numbers[n+1-k]-numbers[k];
	if(m<2)
		cout<<"NO"<<endl<<m;
	else
	{
		if(zhishu(m))
			cout<<"YES"<<endl<<m;
		else
			cout<<"NO"<<endl<<m;
	}
	#ifdef FILE_IO
		fclose(stdin);
		fclose(stdout);
	#endif
	return 0;
} 

int zhishu(long long x)
{
	for(long long i=2;i*i<=x;i++)
	{
		if(x%i==0)
			return 0;
	}
	return 1;
}