【LeetCode】204. Count Primes

问题描述

Description:

Count the number of prime numbers less than a non-negative number, n.

点击查看问题

问题分析

筛选法求素数,参考这篇文章【算法】普通方法和筛选法求素数,这道题用普通方法求素数会超时,只能用筛选法。

代码

class Solution {
public:
    int countPrimes(int n) {
    	if(n<=1) return 0;
    	n--;
    	bool val[n];
    	for(int i=0;i<n;i++) {
	    	val[i] = true;
	    }
		val[0] = false;
		for(int i=1;i<n;i++) {
			if(val[i]==false) continue; 
			// 判断 i+1 是否为素数
			int tmp = i+1;
			bool isPrime = true;
			for(int j=2;j*j<=tmp;j++) {
				if(tmp%j==0) {
					// 不是素数 
					isPrime = false;
					break;
				}
			}
			if(isPrime) {
				// 将这个素数所有的倍数都设置为 false
				for(int j=tmp+tmp;j<=n;) {
					val[j-1] = false;
					j+=tmp;
				} 
			}
		}
		int count = 0;
		for(int i=0;i<n;i++) {
			if(val[i]) {
				count++;
			}
		}
    	return count++;
    }
};