結果
| 問題 |
No.1419 Power Moves
|
| コンテスト | |
| ユーザー |
rlangevin
|
| 提出日時 | 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])
rlangevin