[Tyvj1001]第K極值 (貪心?類比)

標籤：為我   sqrt   main   輸入   lap   ide   using   class   close   

考前打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 21 2 3 4 5

輸入範例

YES2

輸出範例

因為我太菜了，取第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極值 (貪心?類比)