LeetCode 204: Count Primes

LeetCode 204: Count Primes

题目描述

给定一个非负整数，计算小于该整数的所有素数（最小的素数为2）。

解题思路

该题的解法有一个特定的名字：埃拉托斯特尼筛法，下面简单说一下，具体可以参阅网上的资料。

判断一个数x是否是合数，只要依次除以2至sqrt(x)间的素数，判断是否整除即可。

埃拉托斯特尼筛法基于以下原理，给定一个素数n>1,kn是一个合数(k>1)，例如n=2,那么4,6,8,10…都是合数。

以100为例，我们先创建一个100个数字的数组。 

先使用最小的素数2，将所有2的倍数(除2本身)标记为合数。 
接下来2+1的数是3，此时检查3是不是素数，检查标记，发现没有被标记为合数(因为不是2的倍数)，所以再将所有3的倍数标记为合数。 
下一个数是4，发现他已经被标记为合数，所以他可以表示小于4的素数的乘积2*2，所以4的倍数必定含有因子2，所以所有4的倍数已经全部被标记过，直接跳过4。 
下一个数是5，没有被标记为合数，把所有小于100的5的倍数标记为合数.
这样一直计算到sqrt(100)，即10。 

AC代码

class Solution {
public:
	int countPrimes(int n) {
		if (n <= 2)
			return 0;
		vector<bool> isPrimers(n + 1, true);
		int s = sqrt(n);
		for (int i = 2; i <= s; ++i)
		{
			if (isPrimers[i]) //如果当前为素数
			{
				for (int j = i * i; j <= n; j += i)
					isPrimers[j] = false; //将当前素数的倍数标记为合数
			}

		}
		//统计素数的个数
		int res = count_if(isPrimers.begin(), isPrimers.end(), [](bool a) {return a == true; });
		res -= 2; //去掉0和1
		if (isPrimers[n]) //如果n为合数，则去掉n
			res -= 1;
		return res;
	}
};