LeetCode 204 计数质数

题目

代码实现：

方法一（暴力法）：

class Solution {
    public int countPrimes(int n) {
        if(n <= 2){
            return 0;
        }
        if(n == 3){
            return 1;
        }
        if(n == 4){
            return 2;
        }//对于，i小于4时，下面循环体是不会执行的，所以需要分开讨论
        int count = 2;
        for(int i = 4;i < n;i++){
            int a = (int)Math.sqrt(i);
            boolean flag = false;
            for(int j = 2;j<=a;j++){
                if(i%j != 0){//如果9%2!=0,符合if条件。9%3=0，已经不是质数，可以break停止判断。（假设j还是小于a）
                    flag = true;                            
                }else{
                    flag = false;
                    break;
                }
            }
            if(flag){
                count++;
            }
        }
        return count;
    }
}

结果：

for循环表达式，改为i=5，i = i+2,可以把偶数先过滤掉。但是优化不理想，就快了10ms。

class Solution {
    public int countPrimes(int n) {
        if(n <= 2){
            return 0;
        }
        if(n == 3){
            return 1;
        }
        if(n == 4){
            return 2;
        }
        int count = 2;
        for(int i = 5;i < n;i=i+2){
            int a = (int)Math.sqrt(i);
            boolean flag = false;
            for(int j = 2;j<=a;j++){
                if(i%j != 0){
                    flag = true;                            
                }else{
                    flag = false;
                    break;
                }
            }
            if(flag){
                count++;
            }
        }
        return count;
    }
}

结果：

方法二（埃氏筛）：

class Solution {
    public int countPrimes(int n) {
        int[] isPrim = new int[n];
        Arrays.fill(isPrim, 1);
        // 从 2 开始枚举到 sqrt(n)。
        for (int i = 2; i * i < n; i++) {
            // 如果当前是素数
            if (isPrim[i] == 1) {
                // 就把从 i*i 开始，i 的所有倍数都设置为 0。
                for (int j = i * i; j < n; j+=i) {
                    isPrim[j] = 0;
                }
            }
        }
        // 计数
        int count = 0;
        for (int i = 2; i < n; i++) {
            if (isPrim[i] == 1) {
                count++;
            }
        }
        return count;
    }
}

结果：