結果
問題 | No.1419 Power Moves |
ユーザー |
![]() |
提出日時 | 2021-03-05 23:17:00 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 69 ms / 2,000 ms |
コード長 | 1,310 bytes |
コンパイル時間 | 158 ms |
コンパイル使用メモリ | 82,216 KB |
実行使用メモリ | 77,236 KB |
最終ジャッジ日時 | 2024-10-07 04:52:58 |
合計ジャッジ時間 | 3,676 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
# for n in range(5): # kouho = [] # for i in range(1 << n): # res = 0 # for j in range(n): # if (i >> j) & 1: # res += pow(2, j) # else: # res -= pow(2, j) # kouho.append(res) # print(kouho) mod = 10 ** 9 + 7 n, k = map(int, input().split()) ans = [0] * n if n % 2: # 奇数 rui = pow(2, k - 1, n * mod) num, rem = rui // n, rui % n for i in range(n): temp = (2 * i + 1) % n ans[temp] += 1 ans[-temp] += 1 for i in range(n): ans[i] *= num for i in range(rem): temp = (2 * i + 1) % n ans[temp] += 1 ans[-temp] += 1 inv = pow(2, k, mod) inv = pow(inv, mod - 2, mod) for a in ans: res = a % mod res *= inv print(res % mod) else: # 奇数 rui = pow(2, k - 1, n * mod) num, rem = rui // n, rui % n for i in range(n): temp = (2 * i + 1) % n ans[temp] += 1 ans[-temp] += 1 for i in range(n): ans[i] *= num for i in range(rem): temp = (2 * i + 1) % n ans[temp] += 1 ans[-temp] += 1 inv = pow(2, k, mod) inv = pow(inv, mod - 2, mod) for a in ans: res = a % mod res *= inv print(res % mod)