PAT乙级：1007

解决1007 素数对猜想超时（JAVA)

让我们定义d n为：dn=pn+1−pn，其中pi是第i个素数。显然有d1=1，且对于n>1有dn是偶数。“素数对猜想”认为“存在无穷多对相邻且差为2的素数”。
现给定任意正整数N(<105)，请计算不超过N的满足猜想的素数对的个数。

输入格式:
输入在一行给出正整数N。

输出格式:
在一行中输出不超过N的满足猜想的素数对的个数。

---

输入样例:
20
输出样例:
4

---

代码

第一次提交时总是提示运行超时

import java.util.ArrayList;
import java.util.Scanner;

public class p1007 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int m = sc.nextInt();
        if(m==2||m==1){
            System.out.println(0);
        }else{
        int[] arr = new int[10000];
        arr[0]=2;
        int z=0;
        int n=1;
        for (int i=3;i<=m;i++) {
            for (int j = 2; j <=i/2+1; j++) {
                if (i % j == 0){
                    break;
                }
                if(j==i/2+1){
                    arr[n++]=i;
                    z++;
                }
            }
        }
            System.out.println(z);
        int sum =0;
        for (int i=0;i<z+1;i++){
            if(arr[i+1]-arr[i]==2){
                sum++;
            }
        }
        System.out.println(sum);
    }}}

优化后的代码

i++ 变成 i+=2
循环结束条件从 i/2+1 到 Math.sqrt(a)减少复杂度
耗时164ms
提交第二次才答案正确

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        if(n==2||n==1){
            System.out.println(0);
        }else{
        int arr[] = new int[10000];
        arr[0]=2;
        int sum=1;
        for (int i=3;i<=n;i+=2){
            boolean b =isPrime(i);
            if(b==true){
                arr[sum++]=i;
            }
        }
        int m=0;
        for (int i=0;i<sum+1;i++){
            if(arr[i+1]-arr[i]==2){
                m++;
            }
        }

        System.out.println(m);

    }}
    static boolean isPrime(int a){
        for (int i=2;i<=(int)Math.sqrt(a);i++){
            if(a%i==0){

                return false;
            }
        }return true;
    }
}

总结

java太耗时了，几乎每一个都在150ms以上,可能与jvm有关

还可以试试这个

用BufferReader类和StringTokenizer类代替Scanner类来读取输入