204. Count Primes(python+cpp)

题目:

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

Input: 10 
Output: 4 
Explanation: There are 4 prime numbers less than 10, they are 2, 3, 5, 7.

解释:
还是用动态规划的方式做,比写函数开根号的节省时间。
python代码:

from math import sqrt
class Solution(object):
    def countPrimes(self, n):
        """
        :type n: int
        :rtype: int
        """
        
        if n<=2:
            return 0
        isPrime=[True]*n
        isPrime[0]=isPrime[1]=False
        # 在选择除数时候的一个小技巧.大于一半的数是不可能做除数的.可以节省时间
        for i in xrange(2,int(n**0.5)+1):
            if isPrime[i]==True:
                '''
                j=i*i
                while j<n:
                    isPrime[j]=False
                    j+=i
                '''
                isPrime[i * i: n: i] = [False] * len(isPrime[i * i: n: i])  # 非常简洁的语句
        return sum(isPrime)

c++代码:

#include <cmath>
#include <vector>
#include <algorithm>
using namespace std;
class Solution {
public:
    int countPrimes(int n) {
        vector<bool> isPrime(n,true);
        isPrime[0]=false;
        isPrime[1]=false;
        for (int i=2;i<int(sqrt(n))+1;i++)
        {
            if(isPrime[i])
            {
                int j=i*i;
                while (j<n)
                {
                    isPrime[j]=false;
                    j+=i;
           
                }
            } 
        }
        return count(isPrime.begin(),isPrime.end(),true);   
    }
};

总结:
也是再做的时候才发现j需要遍历到i**2+1就好了,这样写,时间变了原来的1/3,震惊。
python可以直接sum()一个bool数组来求数组中的True的个数,用python的切片实现赋值简直震惊,注意数组初始化为True
注意题目是less than n,所以数组元素初始化为n

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值