結果

問題 No.2751 429-like Number
ユーザー mottchanmottchan
提出日時 2024-05-10 23:25:27
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 1,339 bytes
コンパイル時間 264 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 17,692 KB
最終ジャッジ日時 2024-05-10 23:25:34
合計ジャッジ時間 7,080 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 26 ms
10,880 KB
testcase_01 AC 29 ms
10,752 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 AC 27 ms
10,880 KB
testcase_05 AC 26 ms
10,880 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 129 ms
10,880 KB
testcase_09 AC 303 ms
10,880 KB
testcase_10 TLE -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

def suspect(a, t, n):
    x = pow(a, t, n)
    n1 = n - 1
    while t != n1 and x != 1 and x != n1:
        x = pow(x, 2, n)
        t <<= 1
    return t & 1 or x == n1
 
# メイン
# 2^64までの決定的アルゴリズムとして実装しているので、ランダム要素は無い
# ランダムとして用いる場合は、check_listにランダム抽出された数を採用し、20～50個程度試す
def miller_rabin(n):
    if n == 2:
        return True
    if n < 2 or n % 2 == 0:
        return False
    d = (n - 1) >> 1
    while d & 1 == 0:
        d >>= 1
    check_list = (2, 7, 61) if n < 2 ** 32 else (2, 325, 9375, 28178, 450775, 9780504, 1795265022)
    for i in check_list:
        if i >= n:
            break
        if not suspect(i, d, n):
            return False
    return True

def prime_factorize(n):
    a = []
    cnt=0
    while n % 2 == 0 and cnt<2:
        a.append(2)
        n //= 2
        cnt+=1
    f = 3
    while f * f <= n and cnt<2:
        if n % f == 0:
            a.append(f)
            n //= f
            cnt+=1
        else:
            f += 2

    return a

q = int(input())

for _ in range(q):
    n = int(input())
    l = prime_factorize(n)
    # print(l,len(l))
    if len(l)==2:
        if miller_rabin(n//(l[0]*l[-1])):
            print('Yes')
    else:
        print('No')
0