运行会超时,希望大家可以帮助改进
import java.util.Scanner;
public class Main{
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
// 接收正整数N,回车结束
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++) {
// 设置标签,假设是素数
// 判断素数
boolean flag = true;
if (a[i] == 0 || a[i] == 1) {
System.out.println("No");
continue;
}
for (int j = 2; j < a[i]; j++) {
// 如果是不是素数
if (a[i] % j == 0) {
flag = false;
System.out.println("No");
break;
}
}
if (flag) {
System.out.println("Yes");
}
}
}
}