結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 48 ms
56,192 KB
testcase_01 AC 48 ms
57,148 KB
testcase_02 AC 46 ms
55,984 KB
testcase_03 AC 47 ms
56,732 KB
testcase_04 AC 46 ms
55,968 KB
testcase_05 AC 49 ms
61,132 KB
testcase_06 AC 49 ms
61,260 KB
testcase_07 AC 51 ms
60,972 KB
testcase_08 AC 50 ms
61,768 KB
testcase_09 AC 47 ms
55,200 KB
testcase_10 AC 49 ms
62,188 KB
testcase_11 AC 46 ms
56,688 KB
testcase_12 AC 46 ms
55,724 KB
testcase_13 AC 46 ms
56,152 KB
testcase_14 AC 50 ms
61,884 KB
testcase_15 AC 46 ms
56,192 KB
testcase_16 AC 46 ms
55,324 KB
testcase_17 AC 46 ms
56,196 KB
testcase_18 AC 50 ms
62,372 KB
testcase_19 AC 46 ms
56,264 KB
testcase_20 AC 47 ms
56,428 KB
testcase_21 AC 47 ms
56,360 KB
testcase_22 AC 47 ms
57,032 KB
testcase_23 AC 46 ms
55,788 KB
testcase_24 AC 46 ms
55,592 KB
testcase_25 AC 46 ms
56,264 KB
testcase_26 AC 47 ms
56,772 KB
testcase_27 AC 47 ms
55,932 KB
testcase_28 AC 46 ms
55,648 KB
testcase_29 AC 45 ms
56,400 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