LeetCode 204. 计数质数（C++）

统计所有小于非负整数 n 的质数的数量。

示例:

输入: 10
输出: 4
解释: 小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。

假设x是整数，那么只有6*x-1或者6*x+1才可能是质数

C++

class Solution {
public:
    bool judge(int m)
    {
        if(m<=1)
        {
            return false;
        }
        else if(2==m || 3==m)
        {
            return true;
        }
        else
        {
            if(0==(m+1)%6 || 0==(m-1)%6)
            {
                int k=(int)sqrt(m);
                for(int i=2;i<=k;i++)
                {
                    if(0==m%i)
                    {
                        return false;
                    }
                }
                return true;                
            }
            else
            {
                return false;
            }
        }
    }
    
    int countPrimes(int n) 
    {
        if(0==n || 1==n)
        {
            return 0;
        }
        int res=0;
        for(int i=2;i<n;i++)
        {
            if(judge(i))
            {
                res++;
            }
        }
        return res;
    }
};