考前打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;
}

[Tyvj1001]第K極值 (貪心?模擬)