1979 第K个数

1979 第K个数

 

时间限制: 1 s

空间限制: 1000 KB

题目等级 : 黄金 Gold

 

 

 

 

题目描述 Description

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

输入描述 Input Description

第一行为2个数n，k（含义如上题）
第二行为n个数，表示这个序列

输出描述 Output Description

如果m为质数则
第一行为'YES'(没有引号）
第二行为这个数m
否则 
第一行为'NO'
第二行为这个数m

样例输入 Sample Input

5 2
1 2 3 4 5

样例输出 Sample Output

YES
2

数据范围及提示 Data Size & Hint

20%数据满足0<n<=10
50%数据满足0<n<=5000
100%数据满足0<n<=10000

分类标签 Tags 点此展开

sort大法好！

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#define lli long long int
using namespace std;
const int MAXN=100001;
lli a[MAXN];
int comp(const int & a,const int & b)
{
    return a>b;
}
int pd(int n)
{
    if(n<0)
    {
        return 0;
    }
    for(int i=2;i<=sqrt(n);i++)
    {
        if(n%i==0)
        return 0;
    }
    return 1;
}
int main()
{
    lli n,m;
    //scanf("%lld%lld",&n,&m);
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
    }
    sort(a+1,a+n+1,comp);
    lli maxn=a[m];
    sort(a+1,a+n+1);
    lli minn=a[m];
    lli ans=maxn-minn;
    if(pd(ans)==1)
    {
        printf("YES\n%lld",ans);
    }
    else
    {
        printf("NO\n%lld",ans);
    }
    return 0;
}