【LittleXi】素数の数え方

【LittleXi】素数の数え方

トピックの説明

整数 N を指定すると、非負の整数 N より小さいすべての素数の数を返します。
204. 素数を数える
P3912 素数の数

問題解決のアイデア

エラトステネスのふるい

各数値を計算するときは、素数の倍数をマークします。

# 8112ms
class Solution:
    def countPrimes(self, n: int) -> int:
        cnt=0
        f=[1]*n
        for i in range(2,n):
            if f[i]:
                cnt+=1
                j=1
                while i*j<n:
                    f[i*j]=0
                    j+=1
        return cnt

オイラーふるい

最適化されたバージョンでは、素数をマークするときに、繰り返しマークされたものはマークされなくなります。

#python版本超时
class Solution:
    def countPrimes(self, n: int) -> int:
        prime=[0]*(n+1)
        vis=[1]*n
        for i in range(2,n):
            if vis[i]:
                vis[i]=0
                prime[0]+=1
                prime[prime[0]]=i
            for j in range(1,prime[0]+1):
                if i*prime[j]>n:
                    break
                if i*prime[j]<n:
                    vis[i*prime[j]]=0
                if i%prime[j]==0:
                    break
        return prime[0]

//cpp版本272ms
class Solution {
    
    
public:
    int countPrimes(int n) {
    
    
        int vis[5000010] = {
    
     0 };
        int prime[5000010] = {
    
     0 };
        for (int i = 2; i < n; i++)
        {
    
    
            if (vis[i] == 0)
            {
    
    
                vis[i] = 1;
                prime[++prime[0]] = i;
            }
            for (int j = 1; prime[j] * i <= n &&j <= prime[0]; j++)
            {
    
    
                vis[i * prime[j]] = 1;
                if (i % prime[j]==0)
                    break;
            }           
        }
        return prime[0];
    }
};