1. 程式人生 > >LeetCode: 204. Count Primes判斷是否為素數的高效方法

LeetCode: 204. Count Primes判斷是否為素數的高效方法

連結: https://leetcode.com/problems/count-primes/#/description
Description:
Count the number of prime numbers less than a non-negative number, n.
計算小於n的素數的個數。
這道題用普通方法來判斷素數的話必然超限,在別人的部落格上看到了一個比較高效的判斷素數的方法: 大於等於5的質數一定和6的倍數相鄰。例如5和7,11和13,17和19等等;換句話說:不與6的倍數相鄰的數一定不是素數,與6的倍數相鄰的數不一定是素數。所以我們先排除掉不與6相鄰的數,再判斷與6的倍數相鄰的數是否是素數。程式碼如下:

 int countPrimes(int n) {
        int sum = 0;
        if(n==0 || n==1) return 0;
        for(int i=2; i<n; i++){
            if(isPrime(i)) sum++;
        }
        return sum;
    }
    bool isPrime(int num){
                 if(num ==2|| num==3 )  
                                 return true
; //不在6的倍數兩側的一定不是質數 if(num %6!= 1&&num %6!= 5) return false ; int tmp =sqrt( num); //在6的倍數兩側的也可能不是質數 for(int i= 5;i <=tmp; i+=6 ) if
(num %i== 0||num %(i+ 2)==0 ) return false ; //排除所有,剩餘的是質數 return true ; }