結果
問題 |
No.1419 Power Moves
|
ユーザー |
![]() |
提出日時 | 2023-11-30 12:44:47 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 85 ms / 2,000 ms |
コード長 | 482 bytes |
コンパイル時間 | 271 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 77,312 KB |
最終ジャッジ日時 | 2024-09-26 13:56:30 |
合計ジャッジ時間 | 4,926 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
from math import gcd N, K = map(int, input().split()) mod = 10 ** 9 + 7 ans = [0] * N v = pow(2, K, mod) g = gcd(N, 2) Ng = N // g r = pow(2, K, N) q = (v - r) * pow(Ng, mod - 2, mod) % mod now = (pow(2, K, N) - 1) % N for i in range(N): if (i - now) % g == 0: ans[i] += q # print(now, v, q, r) for i in range(r): ans[now] += 1 now = (now - 2) % N # print(ans) inv = pow(v, mod - 2, mod) for i in range(N): ans[i] *= inv ans[i] %= mod print(ans[i])