Java 1094 谷歌的招聘 (20 分) 解决测试点6超时

7-3 谷歌的招聘 (20 分)
2004 年 7 月，谷歌在硅谷的 101 号公路边竖立了一块巨大的广告牌（如下图）用于招聘。内容超级简单，就是一个以 .com 结尾的网址，而前面的网址是一个 10 位素数，这个素数是自然常数 e 中最早出现的 10 位连续数字。能找出这个素数的人，就可以通过访问谷歌的这个网站进入招聘流程的下一步。

本题要求你编程解决一个更通用的问题：从任一给定的长度为 L 的数字中，找出最早出现的 K 位连续数字所组成的素数。

输入格式：
输入在第一行给出 2 个正整数，分别是 L（不超过 1000 的正整数，为数字长度）和 K（小于 10 的正整数）。接下来一行给出一个长度为 L 的正整数 N。

输出格式：
在一行中输出 N 中最早出现的 K 位连续数字所组成的素数。如果这样的素数不存在，则输出 404。注意，原始数字中的前导零也计算在位数之内。例如在 200236 中找 4 位素数，0023 算是解；但第一位 2 不能被当成 0002 输出，因为在原始数字中不存在这个 2 的前导零。

输入样例 1：
20 5
23654987725541023819

输出样例 1：
49877

输入样例 2：
10 3
2468001680

输出样例 2：
404

这个是测试点6超时的代码：

**错误代码👇：**因为在判断是否为素数的时候超时了。要改进一下

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int L = scanner.nextInt();
        int K = scanner.nextInt();
        String st = scanner.next();
        int flag =0;
        for(int i=0;i+K<=st.length();i++){
                int num = Integer.parseInt(st.substring(i,i+K));
                if(isPrime(num)){
                    System.out.println(st.substring(i,i+K));
                    flag=1;
                    break;
                }
        }
        if(flag==0){
            System.out.println("404");
        }
    }
    public static boolean isPrime(int n){
        if(n ==1  || n==0){
            return false;
        }
        for (int i=2;i<n;i++){
            if(n %i ==0){
                return  false;
            }
        }
        return true;
    }
}

正确代码👇：

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int L = scanner.nextInt();
        int K = scanner.nextInt();
        String st = scanner.next();
        int flag = 0;
        if (K > L) {
            System.out.println("404");
        } else {
            for (int i = 0; i + K <= st.length(); i++) {
                int num = Integer.parseInt(st.substring(i, i + K));
                if (isPrime(num)) {
                    System.out.println(st.substring(i, i + K));
                    flag = 1;
                    break;
                }


            }
            if (flag == 0) {
                System.out.println("404");
            }
        }

    }

    public static boolean isPrime (int n) {
        if (n == 1 ||  n == 0) {
            return false;
        }
        for (int i = 2; i*i <= n; i++) {
            if (n % i == 0) {
                return false;
            }
        }
        return true;
    }
}

对于每个数n，其实并不需要从2判断到n-1，我们知道，一个数若可以进行因数分解，那么分解时得到的两个数一定是一个小于等于sqrt(n)，一个大于等于sqrt(n)，据此，上述代码中并不需要遍历到n-1，遍历到sqrt(n)即可.
Java判断是否为素数