結果

問題 No.36 素数が嫌い!
コンテスト
ユーザー Masahiro H
提出日時 2016-02-29 21:51:44
言語 PyPy2
(7.3.15)
結果
RE  
実行時間 -
コード長 506 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 151 ms
コンパイル使用メモリ 77,764 KB
最終ジャッジ日時 2025-12-03 19:34:34
ジャッジサーバーID
(参考情報)
judge2 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2 WA * 1 RE * 1
other AC * 6 WA * 3 RE * 17
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

def is_prime(q,k=50):
    q = abs(q)

    if q == 2: return True
    if q < 2 or q&1 == 0: return False

    d = (q-1)>>1
    while d&1 == 0:
        d >>= 1

    for i in xrange(k):
        a = random.randint(1,q-1)
        t = d
        y = pow(a,t,q)

        while t != q-1 and y != 1 and y != q-1: 
            y = pow(y,2,q)
            t <<= 1
        if y != q-1 and t&1 == 0:
            return False
    return True

d = int(raw_input())

if is_prime(d):
    print("NO")
else:
    print("YES")

0