关于筛法求素数（质数）问题 LeetCode 204.计算质数

• 引用

204. 计数质数
Grubbyskyer博客

• 正文
  找素数或者找质数是算法题中最基础的题型之一。方法同样众多，不同的是效率不相同。从最基础的双层n循环到开方循环再到筛法，效率不断提高，这里我们便省略最基础的两种双层循环的方法，直接从筛法说起。
  • 埃氏筛法
    埃氏筛法的主要思想为，大于1的任意数n的倍数m都不是素数，因为m必然可以被n整除。
    利用这个思路，我们可以建立一个足够大的标记数组，当检测到一个数i后，将i的倍数全部置成一个标记，等判断到这些数的时候便不需要使用循环整除的办法判断，而直接可以判断。

  import java.util.ArrayList;
  import java.util.Arrays;
  import java.util.List;
  
  class Solution {
  
      public int countPrimes(int n) {
          if (n <= 1) return 0;
          boolean[] isPrimesArray = new boolean[n];
          Arrays.fill(isPrimesArray, true);
          isPrimesArray[0] = isPrimesArray[1] = false;
          List<Integer> primeList = new ArrayList<>();
  
          for (int i = 2; i < n; ++i) {
              if (isPrimesArray[i]) {
                  primeList.add(i);
              }
              setNotPrimes(isPrimesArray, i);
          }
          
          return primeList.size();
      }
      
      public void setNotPrimes(boolean[] isPrimesArray, int n) {
          int tmp = n + n;
          for (int i = tmp; i < isPrimesArray.length; i += n) {
              isPrimesArray[i] = false;
          }
      }
  }
  

  • 线性筛法
    上述筛法可以很有效的提高时间效率，在LeetCode204题中耗时100ms左右，如果使用数组保存素数而不是List，可以提高到90ms左右。但是本文主要讲的是筛法所以没有苛求这道题的时间效率。
    埃氏筛法可以提高不少效率但是仔细观察发现还是会有浪费的资源，当检测到数i后，我们会对i的倍数进行筛除，但是i的倍数可能同样是j的倍数，当我们的程序检测到j后，依然再次对这个倍数进行了筛除，所以我们想一个办法，不让一个数被重复筛除。
    我们只需要将当前数的素数倍的素数置标志即可避免重复问题。

  import java.util.ArrayList;
  import java.util.Arrays;
  import java.util.List;
  
  class Solution {
  
      public int countPrimes(int n) {
          if (n <= 1) return 0;
          boolean[] isPrimesArray = new boolean[n];
          Arrays.fill(isPrimesArray, true);
          isPrimesArray[0] = isPrimesArray[1] = false;
          List<Integer> primeList = new ArrayList<>();
  
          for (int i = 2; i < n; ++i) {
              if (isPrimesArray[i]) {
                  primeList.add(i);
              }
              setNotPrimes(isPrimesArray, i, primeList);
          }
          
          return primeList.size();
      }
      
      public void setNotPrimes(boolean[] isPrimesArray, int n, List<Integer> primeList) {
          for (Integer i : primeList) {
              if (n * i >= isPrimesArray.length) break;
              isPrimesArray[n * i] = false;
          }
      }
  }
  

• 总结
  本文主要针对筛法，而不是针对LeetCode，本文最终方法可以将时间降低到70ms左右，但是并不是最快的，这道题并不要求保存质数，而只需要数量，所以保存质数的操作浪费的大量时间，其实这道题用普通埃氏筛法不保存质数组，可以提升到20ms左右。
  使用筛法时一定要注意，要从2开始筛，如果你只需要x到y之间的所有素数，那就要考虑是否需要筛法，如果使用筛法有两种选择：
  这两种方法需要根据输入的长度以及所需的数据来判断效率：
  • 检测到i时，除了使用标志数组，还要对i是否为素数进行真实判断。因为不是从2开始筛除，所以标志数组保存的仅仅为肯定不是素数的数，而不是一定为素数的数。判断可以使用开方循环的方法。

  public boolean isPrimes(int n) {
      for (int i = 2; i < Math.sqrt(n); ++i) {
          if (n % i == 0) return false;
      }
      
      return true;
  }
  

  • 从2开始检测，但是最后取值的时候取你需要的值。