結果
問題 |
No.1258 コインゲーム
|
ユーザー |
|
提出日時 | 2024-01-21 23:32:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 357 ms / 2,000 ms |
コード長 | 783 bytes |
コンパイル時間 | 338 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 80,568 KB |
最終ジャッジ日時 | 2024-09-28 06:17:01 |
合計ジャッジ時間 | 16,476 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 50 |
ソースコード
## https://yukicoder.me/problems/no/1258 MOD = 10 ** 9 + 7 def solve(N, M, X): if M != 1: a = pow(1 + M, N, MOD) b = pow(1 - M, N, MOD) inv2 = pow(2, MOD - 2, MOD) if X == 0: ans = (a + b) % MOD ans *= inv2 ans %= MOD else: ans = (a - b) % MOD ans *= inv2 ans %= MOD return ans else: a = pow(1 + M, N, MOD) inv2 = pow(2, MOD - 2, MOD) ans = a * inv2 return ans % MOD def main(): S = int(input()) answers = [] for _ in range(S): N, M, X = map(int, input().split()) answers.append(solve(N, M, X)) for ans in answers: print(ans) if __name__ == "__main__": main()