結果
問題 |
No.1810 RGB Biscuits
|
ユーザー |
![]() |
提出日時 | 2022-07-27 13:29:04 |
言語 | Java (openjdk 23) |
結果 |
AC
|
実行時間 | 137 ms / 2,000 ms |
コード長 | 2,439 bytes |
コンパイル時間 | 3,359 ms |
コンパイル使用メモリ | 77,836 KB |
実行使用メモリ | 41,376 KB |
最終ジャッジ日時 | 2024-07-16 19:11:19 |
合計ジャッジ時間 | 5,784 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 20 |
ソースコード
import java.io.*; import java.util.*; public class Main { static final int MOD = 1000000007; public static void main(String[] args) throws Exception { Scanner sc = new Scanner(); int a = sc.nextInt(); int b = sc.nextInt(); long[][][] matrix = new long[60][3][3]; matrix[0] = new long[][]{{a, b, 0}, {1, 0, 0}, {0, 0, 0}}; for (int i = 1; i < 60; i++) { for (int e = 0; e < 3; e++) { for (int f = 0; f < 3; f++) { for (int d = 0; d < 3; d++) { matrix[i][e][d] += matrix[i - 1][e][f] * matrix[i - 1][f][d] % MOD; matrix[i][e][d] %= MOD; } } } } int n = sc.nextInt(); StringBuilder sb = new StringBuilder(); while (n-- > 0) { long x = sc.nextLong(); boolean hasNext = (x % 2 == 1); x /= 2; long[] ans = new long[]{1, 1, 0}; for (int i = 59; i >= 0; i--) { if (x < (1L << i)) { continue; } x -= (1L << i); long[] next = new long[3]; for (int e = 0; e < 3; e++) { for (int f = 0; f < 3; f++) { next[e] += ans[f] * matrix[i][e][f] % MOD; next[e] %= MOD; } } ans = next; } if (hasNext) { sb.append((ans[0] * (a + 1) + ans[1] * (b + 1)) % MOD).append("\n"); } else { sb.append((ans[0] + ans[1]) % MOD).append("\n"); } } System.out.print(sb); } } class Scanner { BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); StringTokenizer st = new StringTokenizer(""); public Scanner() throws Exception { } public int nextInt() throws Exception { return Integer.parseInt(next()); } public long nextLong() throws Exception { return Long.parseLong(next()); } public double nextDouble() throws Exception { return Double.parseDouble(next()); } public String next() throws Exception { while (!st.hasMoreTokens()) { st = new StringTokenizer(br.readLine()); } return st.nextToken(); } }