結果
問題 |
No.1810 RGB Biscuits
|
ユーザー |
|
提出日時 | 2022-01-14 23:18:36 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 615 ms / 2,000 ms |
コード長 | 547 bytes |
コンパイル時間 | 95 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 44,236 KB |
最終ジャッジ日時 | 2024-11-20 14:25:58 |
合計ジャッジ時間 | 14,732 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 20 |
ソースコード
import numpy as np A, B = map(int, input().split()) n = int(input()) MOD = 10 ** 9 + 7 M1 = np.array([[1, 0, 0], [0, 1, 0], [A, B, 0]]) M2 = np.array([[0, 0, 1], [1, 0, 0], [0, 0, 0]]) M = M2 @ M1 doubling = [M1, M] for i in range(61): doubling.append(doubling[-1] @ doubling[-1]) doubling[-1] = np.mod(doubling[-1], MOD) for i in range(n): T = int(input()) V = np.array([1, 1, 0]) for i in range(61)[::-1]: if T >> i & 1: V = doubling[i] @ V V = np.mod(V, MOD) print(np.sum(V) % MOD)