結果

問題 No.144 エラトステネスのざる
ユーザー Navier_BoltzmannNavier_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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 108 ms
97,892 KB
testcase_01 AC 111 ms
97,500 KB
testcase_02 AC 110 ms
97,536 KB
testcase_03 AC 109 ms
97,780 KB
testcase_04 AC 105 ms
97,316 KB
testcase_05 AC 107 ms
97,280 KB
testcase_06 AC 145 ms
100,940 KB
testcase_07 AC 142 ms
100,832 KB
testcase_08 AC 147 ms
100,836 KB
testcase_09 AC 155 ms
100,736 KB
testcase_10 AC 144 ms
100,696 KB
testcase_11 AC 150 ms
100,736 KB
testcase_12 AC 148 ms
100,864 KB
testcase_13 AC 720 ms
101,608 KB
testcase_14 AC 690 ms
101,484 KB
testcase_15 AC 683 ms
101,248 KB
testcase_16 AC 707 ms
101,608 KB
testcase_17 AC 691 ms
101,496 KB
testcase_18 AC 683 ms
101,248 KB
testcase_19 AC 750 ms
101,340 KB
権限があれば一括ダウンロードができます

ソースコード

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