結果

問題 No.36 素数が嫌い！
ユーザー lam6er
提出日時 2025-03-26 15:43:57
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,569 bytes
コンパイル時間 158 ms
コンパイル使用メモリ 82,772 KB
実行使用メモリ 58,580 KB
最終ジャッジ日時 2025-03-26 15:44:03
合計ジャッジ時間 2,442 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 19 WA * 7
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import math

def is_prime(n):
    if n < 2:
        return False
    for p in [2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37]:
        if n % p == 0:
            return n == p
    d = n - 1
    s = 0
    while d % 2 == 0:
        d //= 2
        s += 1
    for a in [2, 325, 9375, 28178, 450775, 9780504, 1795265022]:
        if a >= n:
            break
        x = pow(a, d, n)
        if x == 1 or x == n - 1:
            continue
        for _ in range(s - 1):
            x = pow(x, 2, n)
            if x == n - 1:
                break
        else:
            return False
    return True

def find_factor(n):
    if n % 2 == 0:
        return 2
    max_i = int(math.isqrt(n)) + 1
    i = 3
    while i <= max_i:
        if n % i == 0:
            return i
        i += 2
    return n  # This line is theoretically unreachable since n is composite

def main():
    N_input = sys.stdin.read().strip()
    N = int(N_input)
    if N == 1:
        print("NO")
        return
    if is_prime(N):
        print("NO")
        return
    p = find_factor(N)
    e = 0
    m = N
    while m % p == 0:
        m //= p
        e += 1
    if m == 1:
        if e >= 3:
            print("YES")
        else:
            print("NO")
    else:
        if is_prime(m):
            if m != p:
                print("YES")
            else:
                total_e = e + 1
                if total_e >= 3:
                    print("YES")
                else:
                    print("NO")
        else:
            print("YES")

if __name__ == "__main__":
    main()
0