PAT乙级：1007 素数对猜想（JAVA）

import java.util.*;
public class Main{

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner reader = new Scanner(System.in);
		int n = reader.nextInt();
		int sum = 0;
		if(n==0 || n==1) {
			System.out.println("0");
			return;
		}
		int[] num = sushu(n);
		for(int i = 0;i<num.length-1;i++) {
//			System.out.println(num[i]);
			int first = num[i];
			int second = num[i+1];
			if(second == 0) {
				break;
			}
			if(second - first == 2) {
				sum++;
			}
		}
		System.out.println(sum);
	}
	public static int[] sushu(int n) {
		int[] num  = new int[n];
		int count=0;
		for(int i=2;i<=n;i++) {
			Boolean flag = true;
			for(int j=2;j<=Math.sqrt(i);j++) {
				if(i%j==0) {
					flag = false;
					break;
				}
			}
			if(flag) {
				num[count++] = i;
			}
		}
		return num;
	}
}

关于测试点5超时，因为在这一块

for(int j=2;j<=Math.sqrt(i);j++) {
				if(i%j==0) {
					flag = false;
					break;
				}
			}

中写成了j<i循环次数过多，导致超时，写成sqrt就不会超时了