結果

問題 No.144 エラトステネスのざる
コンテスト
ユーザー FromBooska
提出日時 2023-02-15 09:54:51
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,467 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 345 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 116,480 KB
最終ジャッジ日時 2024-07-17 20:20:42
合計ジャッジ時間 4,687 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 7 WA * 3 TLE * 1 -- * 6
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

# 素数は常に素数と判定される
# 非素数は、約数の数だけ(1と自身を除く)非素数と判定される可能性がある
# たとえば4は約数2のときに判定されなければ残ってしまう
# たとえば6は約数2と3の2回に判定されなければ残ってしまう
# たとえば8は約数2と4の2回、ところが2は常に素数だが、4は偽判定されている可能性がある
# つまり下から積み上げる必要がある

def divisors(n):
    lower_divisors , upper_divisors = [], []
    i = 1
    while i*i <= n:
        if n % i == 0:
            lower_divisors.append(i)
            if i != n // i:
                upper_divisors.append(n//i)
        i += 1
    return lower_divisors + upper_divisors[::-1]

N, prob = map(float, input().split())
N = int(N)

is_prime = [1]*(N + 1)
is_prime[0] = 0  
is_prime[1] = 0
for p in range(2, N + 1):
    if is_prime[p]:
        for q in range(2*p, N + 1, p):
            is_prime[q] = 0

primes = set()
for i in range(N):
    if is_prime[i] == 1:
        primes.add(i)

#print(primes)

yuki_prime = [0]*(N+1)
# prob = 下線部の処理が行われる確率

for i in range(2, N+1):
    if i in primes:
        yuki_prime[i] = 1
    else:
        divs = divisors(i)[1:-1]
        survive_rate = 1
        for d in divs:
            survive_rate *= (1- yuki_prime[d]*(prob))
        yuki_prime[i] = survive_rate

#print(yuki_prime)

ans = sum(yuki_prime)
print(ans)
0