P1001（Java代码）注意特殊情况

解题思路:  无

注意事项:  注意存在差为0,1的特殊值时，

输入：

5 2

2 2 2 2 2   

输出 ：

NO

0

输入：
5 2

2 2 2 3 3

输出：

NO

1
参考代码:

import java.util.Arrays;
import java.util.Scanner;
public class Main {
	public static void main(String []args) {
		Scanner in =new Scanner(System.in);
		int n=in.nextInt(),k=in.nextInt();
		int arr[]= new int[n];
		for(int i=0;i<n;i++)
			arr[i]=in.nextInt();
		Arrays.sort(arr);
		int tmp=arr[n-k]-arr[k-1];
		boolean flag=true;
		if(tmp<2)
			flag=false;   //注意处理特殊情况
		for(int i=2;i<=Math.sqrt(tmp);i++) {  //判断素数
			if(tmp%i==0) {
				flag=false;
				break;
			}
		}
		if(flag)
			System.out.println("YES");
		else
			System.out.println("NO");
		System.out.println(tmp);
		in.close();
	}
}