Tyvj_P1001 第K极值

最新推荐文章于 2023-01-08 15:32:53 发布

寻找小海螺

最新推荐文章于 2023-01-08 15:32:53 发布

阅读量933

点赞数 1

分类专栏：数据结构和算法

本文链接：https://blog.csdn.net/czkct/article/details/43392085

版权

数据结构和算法专栏收录该内容

160 篇文章 1 订阅

订阅专栏

该博客介绍了如何解决寻找序列中第K小的质数问题，提供了输入和输出格式，并给出了测试样例进行说明。

摘要由CSDN通过智能技术生成

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类型