Tyvj_P1001 第K极值

P1001 第K极值 时间: 1000ms / 空间: 131072KiB / Java类名: Main

背景

成成第一次模拟赛 第一道

描述

给定一个长度为N(0<n<=10000)的序列，保证每一个序列中的数字a[i]是小于maxlongint的非负整数 ，编程要求求出整个序列中第k大的数字减去第k小的数字的值m，并判断m是否为质数。(0<k<=n)

输入格式

输入格式：

第一行为2个数n，k（含义如上题）

第二行为n个数，表示这个序列

输出格式

输出格式：

如果m为质数则

第一行为'YES'(没有引号）

第二行为这个数m

否则 

第一行为'NO'

第二行为这个数m

测试样例1

输入

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

#include<stdio.h>
#include<math.h>
void merge_sort(__int64 *,int ,int ,__int64 *);
bool judge(__int64 );
int main()
{
	int n,k;
	__int64 a[10000],t[10000];
	scanf("%d %d",&n,&k);
	for(int i=0;i<n;i++)
		scanf("%I64d",&a[i]);
	merge_sort(a,0,n,t);
	__int64 m=a[n-k]-a[k-1];
	if(judge(m)) printf("YES\n");
	else printf("NO\n");
	printf("%I64d",m);
	return 0;
}
void merge_sort(__int64 *a,int x,int y,__int64 *t)
{
	if(y-x>1)
	{
		int m=x+(y-x)/2;
		merge_sort(a,x,m,t);
		merge_sort(a,m,y,t);
		int p=x,q=m,i=x;
		while(p<m||q<y)
		{
			if(q>=y||p<m&&a[p]<a[q]) t[i++]=a[p++];
			else t[i++]=a[q++];
		} 
		for(i=x;i<y;i++)
		a[i]=t[i];
	}
}
bool judge(__int64 m)  //素数判断 
{
	if(m<2) return false;  //小心那些小于2的数 
	if(m==2) return true;
	if(m%2==0) return false;
	for(int i=3;i<=sqrt(m);i+=2)
	{
		if(m%i==0) return false;
	}
	return true;
}
//bool  类型 在C中是过不了编译的   但是在C++中可以过的了编译 、、
//C标准中没有bool类型