結果
問題 | No.1419 Power Moves |
ユーザー |
![]() |
提出日時 | 2021-03-05 22:32:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 70 ms / 2,000 ms |
コード長 | 1,836 bytes |
コンパイル時間 | 241 ms |
コンパイル使用メモリ | 82,652 KB |
実行使用メモリ | 78,208 KB |
最終ジャッジ日時 | 2024-10-07 03:11:08 |
合計ジャッジ時間 | 3,958 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
mod = 1000000007 eps = 10**-9 def main(): import sys input = sys.stdin.readline N, K = map(int, input().split()) ans = [0] * N inv2 = pow(pow(2, K, mod), mod-2, mod) if N%2 == 0: pow2_N = (pow(2, K, N) - 1)%N r = (pow2_N + 1) // 2 num = ((pow(2, K-1, mod) - r)%mod * pow((N // 2), mod-2, mod))%mod for i in range(N): if i & 1: if i <= pow2_N: ans[i] = ((num + 1) * inv2)%mod else: ans[i] = (num * inv2)%mod ans_rev = [0] * N for i in range(1, N): ans_rev[i] = ans[-i] for i in range(N): ans[i] = (ans[i] + ans_rev[i])%mod else: pow2_N = (pow(2, K, N) - 1)%N if pow2_N & 1: r = (pow2_N + 1) // 2 else: r = (N+1) // 2 + pow2_N // 2 num = ((pow(2, K - 1, mod) - r) % mod * pow(N, mod - 2, mod)) % mod for i in range(N): if pow2_N & 1: if i & 1: if i <= pow2_N: ans[i] = ((num + 1) * inv2)%mod else: ans[i] = (num * inv2)%mod else: ans[i] = (num * inv2)%mod else: if i&1: ans[i] = ((num + 1) * inv2) % mod else: if i <= pow2_N: ans[i] = ((num + 1) * inv2) % mod else: ans[i] = (num * inv2)%mod ans_rev = [0] * N for i in range(1, N): ans_rev[i] = ans[-i] for i in range(N): ans[i] = (ans[i] + ans_rev[i])%mod ans[0] = (ans[0] * 2)%mod for i in range(N): print(ans[i]) if __name__ == '__main__': main()