LeetCode: 204. Count Primes判斷是否為素數的高效方法
阿新 • • 發佈:2019-01-28
連結: 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 ;
}