P1001 第K极值【tyvj】

/*==========================================
P1001 第K极值
内存限制 128MB     代码限制 64KB
描述 Description
给定一个长度为N(0<n<=10000)的序列，保证每一个序列中的数字a[i]是
小于maxlongint的非负整数 ，编程要求求出整个序列中第k大的数字减去
第k小的数字的值m，并判断m是否为质数。(0<k<=n)
输入格式 InputFormat
输入格式：
第一行为2个数n，k（含义如上题）
第二行为n个数，表示这个序列
输出格式 OutputFormat
输出格式：
如果m为质数则
第一行为'YES'(没有引号）
第二行为这个数m
否则 
第一行为'NO'
第二行为这个数m
样例输入 SampleInput
5 2
1 2 3 4 5
样例输出 SampleOutput
YES
2

数据范围和注释 Hint
对于第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<stdlib.h>
#include<math.h>
int fun(long n);//判断n是否质数，是则返回1，否则返回0； 
int cmp(const void *a,const void *b);
int main()
{
    long n,k,*a,i,m;
    scanf("%ld%ld",&n,&k);
    a=(long *)malloc(n*sizeof(long));
    for(i=0;i<n;i++)
        scanf("%ld",&a[i]);
    qsort(a,n,sizeof(long),cmp);
    m=a[n-k]-a[k-1];
    i=fun(m);
    if(i==1) printf("YES\n%ld\n",m);
    else printf("NO\n%ld\n",m);
    return 0;
}
int fun(long n)//判断n是否质数，是则返回1，否则返回0；
{
    long i,t;
    if(n<2) return 0;
    t=sqrt(n);
    for(i=2;i<=t;i++)
    {
        if(n%i==0) return 0;
    }
    return 1;
}
int cmp(const void *a,const void *b)
{
    return *(long *)a-*(long *)b;
}

这个题呢要用long类型数据。要注意第k大数减去第k小数有可能是负数。质数检测也要注意写正确。可以使用快排的就用快排。

 

好久没写OJ的题目了，再过来看看写写倍觉亲切啊