結果
問題 | No.1255 ハイレーツ・オブ・ボリビアン |
ユーザー |
![]() |
提出日時 | 2020-08-26 00:13:46 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 738 ms / 2,000 ms |
コード長 | 1,036 bytes |
コンパイル時間 | 263 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 45,312 KB |
最終ジャッジ日時 | 2024-07-23 08:42:11 |
合計ジャッジ時間 | 12,225 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 15 |
ソースコード
import sys import numpy as np read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline readlines = sys.stdin.buffer.readlines T, *Ns = map(int, read().split()) def prime_table(N): is_prime = np.zeros(N, np.int64) is_prime[2:3] = 1 is_prime[3::2] = 1 for p in range(3, N, 2): if p * p >= N: break if is_prime[p]: is_prime[p * p::p + p] = 0 return is_prime, np.where(is_prime)[0] is_prime, primes = prime_table(10**5) def euler_phi(N): pf = primes[N % primes == 0] phi = N for p in pf: phi -= phi // p while N % p == 0: N //= p if N > 1: phi -= phi // N return phi def solve(n): n = 2 * n - 1 # 2^k = 1 mod n となる最小の k >= 1 を求める phi_n = euler_phi(n) div = np.arange(1, 10**5 + 1) div = div[phi_n % div == 0] div = np.union1d(div, phi_n // div) for d in div.tolist(): if (pow(2, d, n) -1) % n == 0: return d for n in Ns: print(solve(n))