码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode 204. Count Primes

时间:2015-07-22 16:21:01      阅读:122      评论:0      收藏:0      [点我收藏+]

标签:leetcode   c++   prime   算法   

LeetCode原题维基百科都有解释用到的Sieve of Eratosthenes算法。
该算法可在O(nloglogn)时间内,求出小于n的所有质数;空间复杂度为O(n).
随着n的增大,当空间有限时,维基百科还提出了一种分段筛选(segmented sieve)方法,在时间复杂度不变的情况下,将空间复杂度降为O(n^0.5).

下面代码实现了常规筛选(regular sieve)方法:

class Solution 
{
public:
    int countPrimes(int n) 
    {
        if (n <= 1)
        {
            return 0;
        }

        bool prime[n];
        memset(prime, true, sizeof(prime));
        for (int i = 2; i * i < n; ++ i)
        {
            if (prime[i] == false)
            {
                continue;
            }
            for (int j = i * i; j < n; j += i)
            {
                prime[j] = false;
            }
        }
        return count_if(prime+2, prime+n, [](bool prime) -> bool { return prime;});
    }
};

版权声明:本文为博主原创文章,未经博主允许不得转载。

LeetCode 204. Count Primes

标签:leetcode   c++   prime   算法   

原文地址:http://blog.csdn.net/stephen_wong/article/details/47003203

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!