L1-028 判断素数 java语言

L1-028 判断素数 （10 分）
本题的目标很简单，就是判断一个给定的正整数是否素数。

输入格式：
输入在第一行给出一个正整数N（≤ 10），随后N行，每行给出一个小于2
​31
​​ 的需要判断的正整数。

输出格式：
对每个需要判断的正整数，如果它是素数，则在一行中输出Yes，否则输出No。

输入样例：
2
11
111
输出样例：
Yes
No

import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		int a[] = new int[N];
		for (int i = 0; i < N; i++) {
			a[i] = sc.nextInt();
		}
		for (int i = 0; i < N; i++) {
			sushu(a[i]);
		}
	}
	public static void sushu(int n) {
		int count=0;
		if (n <= 3) {
			System.out.println("Yes");
		} else {
			for (int i = 2; i <= Math.sqrt(n); i++) {
				if (n % i == 0) {
					count++;
					break;
				}
			}
			if(count!=0) {
			System.out.println("No");
			}else {
				System.out.println("Yes");
			}
		}
	}

}