結果

問題 No.2751 429-like Number
ユーザー mottchanmottchan
提出日時 2024-05-10 23:05:32
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 1,508 bytes
コンパイル時間 104 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 18,592 KB
最終ジャッジ日時 2024-05-10 23:05:40
合計ジャッジ時間 8,167 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 48 ms
18,592 KB
testcase_01 AC 46 ms
11,904 KB
testcase_02 AC 50 ms
11,776 KB
testcase_03 AC 50 ms
11,776 KB
testcase_04 AC 47 ms
11,776 KB
testcase_05 AC 46 ms
11,648 KB
testcase_06 AC 53 ms
11,776 KB
testcase_07 AC 1,520 ms
11,776 KB
testcase_08 AC 158 ms
11,648 KB
testcase_09 TLE -
testcase_10 -- -
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 get_prime(n):
    sieve = [True] * (n + 1) 
    i = 2
    while i * i <= n: 
        if sieve[i]:
            for j in range(i * i, n + 1, i):
                sieve[j] = False
        i += 1
    return [i for i in range(2, n + 1) if sieve[i]]

l = get_prime(100000)
# print(l)
q = int(input())

for _ in range(q):
    n = int(input())
    if n==1:
        print('No')
        continue
    cnt=0
    for i in l:
        if n%i == 0:
            n = n//i
            cnt+=1
            break
    for i in l:
        if n%i == 0:
            n = n//i
            cnt+=1
            break
    if miller_rabin(n):
        cnt+=1

    if cnt==3:
        print('Yes')
    else:
        print('No')
0