【tyvj】刷题记录(1001~1099)(64/99)

1001:排序完按照题意做即可。

 1 #include<cstdio>
 2 #include<iostream>
 3 #include<cmath>
 4 #include<algorithm>
 5 using namespace std;
 6 int a[10001],k,n;
 7 int main(){
 8     scanf("%d%d",&n,&k);
 9     for (int i=1; i<=n; i++) scanf("%d",&a[i]);
10     sort(a+1,a+n+1);
11     int m=a[n-k+1]-a[k];
12     if (m<2&&m>=0) printf("NO");
13     else if (m>0){
14         if (m==2) printf("YES\n");
15         else{
16             for (int i=2; i<=floor(sqrt(m)); i++)
17                 if (m%i==0){
18                     printf("NO\n");
19                     printf("%d",m);
20                     return 0;
21                 }
22             printf("YES\n");
23         }
24         printf("%d",m);
25         return 0;
26     }else printf("NO\n%d",m);
27     return 0;
28 }

1001