結果

問題 No.1737 One to N
ユーザー ShirotsumeShirotsume
提出日時 2021-10-11 21:40:30
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 44 ms / 2,000 ms
コード長 1,260 bytes
コンパイル時間 329 ms
コンパイル使用メモリ 82,196 KB
実行使用メモリ 62,224 KB
最終ジャッジ日時 2024-05-04 04:04:52
合計ジャッジ時間 2,568 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
56,408 KB
testcase_01 AC 37 ms
56,536 KB
testcase_02 AC 36 ms
57,612 KB
testcase_03 AC 38 ms
56,904 KB
testcase_04 AC 38 ms
56,612 KB
testcase_05 AC 39 ms
61,052 KB
testcase_06 AC 43 ms
61,476 KB
testcase_07 AC 41 ms
61,316 KB
testcase_08 AC 40 ms
62,048 KB
testcase_09 AC 40 ms
56,652 KB
testcase_10 AC 43 ms
60,768 KB
testcase_11 AC 40 ms
56,264 KB
testcase_12 AC 42 ms
55,524 KB
testcase_13 AC 42 ms
56,068 KB
testcase_14 AC 40 ms
62,224 KB
testcase_15 AC 37 ms
55,956 KB
testcase_16 AC 41 ms
57,164 KB
testcase_17 AC 42 ms
57,064 KB
testcase_18 AC 44 ms
60,836 KB
testcase_19 AC 39 ms
56,668 KB
testcase_20 AC 36 ms
56,096 KB
testcase_21 AC 44 ms
55,960 KB
testcase_22 AC 42 ms
55,928 KB
testcase_23 AC 41 ms
55,900 KB
testcase_24 AC 41 ms
56,720 KB
testcase_25 AC 40 ms
56,920 KB
testcase_26 AC 37 ms
56,716 KB
testcase_27 AC 37 ms
56,100 KB
testcase_28 AC 36 ms
57,252 KB
testcase_29 AC 36 ms
55,644 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import gcd
import random
from collections import defaultdict
def is_prime(n):
    if n==2:
        return True
    if n==1 or n&1==0:
        return False
    d=(n-1)>>1
    while d&1==0:
        d>>=1

    for k in range(100):
        a=random.randint(1,n-1)
        t=d
        y=pow(a, t, n)
        while t!=n-1 and y!=1 and y!=n-1:
            y=(y*y)%n
            t<<=1
        if y!=n-1 and t&1==0:
            return False
    return True

def prime_fact(N):
    res=defaultdict(int)
    if N==1:
        return res
    p=2
    while(p<=10**4 and N>1):
        if N%p==0:
            while(N%p==0):
                res[p]+=1
                N//=p
        p+=1
    while(N>1):
        if is_prime(N):
            res[N]+=1
            break
        x=random.randrange(N)
        y=(x*x+1)%N
        i=1
        while(True):
            d=gcd(abs(x-y),N)
            if d==1:
                i+=1
            elif d==N:
                res[N]+=1
                return res
            else:
                res[d]+=1
                N//=d
                break
            x=(x*x+1)%N
            y=(y*y+1)%N
            y=(y*y+1)%N
    return res

n = int(input())
d = prime_fact(n)
ans = 0
for x, y in d.items():
    ans += x * y
print(ans)
0