結果
問題 | No.36 素数が嫌い! |
ユーザー |
|
提出日時 | 2015-09-25 14:22:41 |
言語 | Python2 (2.7.18) |
結果 |
AC
|
実行時間 | 60 ms / 5,000 ms |
コード長 | 966 bytes |
コンパイル時間 | 93 ms |
コンパイル使用メモリ | 6,784 KB |
実行使用メモリ | 7,680 KB |
最終ジャッジ日時 | 2024-06-27 00:29:22 |
合計ジャッジ時間 | 1,960 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 26 |
ソースコード
import math, itertools, random, collections, time def suspect(a, e, k, n): a = pow(a, e, n) if a == 1: return True for _ in xrange(k): if a == n - 1: return True a = a * a % n return False def isprime(n): if n == 1: return False e = n - 1 k = 0 while e % 2 == 0: e /= 2 k += 1 if n < 2 ** 31: bases = [2, 3, 5, 7] else: bases = [2, 325, 9375, 28178, 450775, 9780504, 1795265022] for i in bases: if n != i and not suspect(i, e, k, n): return False return True def gcd(a, b): if b == 0: return a return gcd(b, a % b) def rho(n): for c in xrange(1, n): if c == n - 2: continue x, y, d = 2, 2, 1 f = lambda x: (x * x + c) % n while d == 1: x, y = f(x), f(f(y)) d = gcd(abs(x - y), n) if d != n: return d return -1 n = input() if n == 1: print "NO" elif isprime(n): print "NO" else: r = rho(n) if r == -1: print "NO" elif isprime(r): if isprime(n / r): print "NO" else: print "YES" else: print "YES"