P1001 （C++代码）水一下

解题思路:
    又水了一发啊
参考代码:

#include<bits/stdc++.h>
#define hh ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
using namespace std;
typedef long int li;
const int maxn=100005;
int n,k,ans;
int a[maxn]={0};
bool is_prime(){
	if(ans<2){
		return false;
	}
	for(int i=2;i<=sqrt(ans);i++){
		if(ans%i==0){
			return false;
		}
	}
	return true;
}
int main(){
	hh;
	cin>>n>>k;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	sort(a+1,a+n+1);
	ans=a[n-k+1]-a[k];
	if(is_prime()){
		cout<<"YES"<<endl;
	}else{
		cout<<"NO"<<endl;
	}
	cout<<ans<<endl;
	return 0;
}