[Tyvj1001]第K极值 (贪心?模拟)

考前打tyvj的水题

题目描述

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

输入格式

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

输出格式

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

测试数据

5 2
1 2 3 4 5

YES
2

因为我太菜了，取第K值只会排序

所以我用排序，然后算出第K大-第K小

然后进行操作

#include<bits/stdc++.h>
using namespace std;
int a[10005];
bool prime(int x)
{
    if(x==1) return true;
    for(int i=2;i<=sqrt(x);i++)
    if(x%i==0) return true;
    return false;
}
int main()
{
    int n,k;scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    sort(a+1,a+1+n);
    int t=a[n-k+1]-a[k];
    if(prime(t)||t<2) puts("NO");
    else puts("YES");
    printf("%d",t);
    return 0;
}