結果

問題 No.144 エラトステネスのざる
コンテスト
ユーザー Navier_Boltzmann
提出日時 2023-11-15 18:40:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 750 ms / 2,000 ms
コード長 1,433 bytes
コンパイル時間 134 ms
コンパイル使用メモリ 82,592 KB
実行使用メモリ 101,608 KB
最終ジャッジ日時 2024-09-26 04:42:15
合計ジャッジ時間 7,580 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import *
from itertools import *
from functools import *
from heapq import *
import sys,math

N,p = input().split()
N = int(N)
p = float(p)


class prime_factorize():
    
    def __init__(self,M=10**6):
        self.sieve = [-1]*(M+1)
        self.sieve[1] = 1
        self.p = [False]*(M+1)
        self.mu = [1]*(M+1)
        
        for i in range(2,M+1):
            if self.sieve[i] == -1:
                self.p[i] = True
                
                i2 = i**2
                for j in range(i2,M+1,i2):
                    
                    self.mu[j] = 0
                
                
                for j in range(i,M+1,i):
                    self.sieve[j] = i
                    
                    self.mu[j] *= -1
                    
    def factors(self,x):
        tmp = []
        while self.sieve[x] != x:
            tmp.append(self.sieve[x])
            x //= self.sieve[x]
        tmp.append(self.sieve[x])
        return tmp
        
    def is_prime(self,x):
        return self.p[x]
        
    def mobius(self,x):
        return self.mu[x]

pf = prime_factorize()

ans = 0

mem = defaultdict(lambda:-1)
mem[0] = 1
def f(x):
    if mem[x]!=-1:
        return mem[x]
    mem[x] = f(x-1)*(1-p)
    return mem[x]




for i in range(2,N+1):
    
    C = Counter(pf.factors(i))
    t = 1
    for v in C.values():
        t *= (1+v)
    # print(i,t)
    ans += f(t-2)
print(ans)
0