結果
問題 | No.718 行列のできるフィボナッチ数列道場 (1) |
ユーザー |
|
提出日時 | 2018-07-27 23:05:45 |
言語 | Java (openjdk 23) |
結果 |
AC
|
実行時間 | 133 ms / 2,000 ms |
コード長 | 1,095 bytes |
コンパイル時間 | 2,081 ms |
コンパイル使用メモリ | 77,376 KB |
実行使用メモリ | 54,236 KB |
最終ジャッジ日時 | 2024-07-05 04:37:54 |
合計ジャッジ時間 | 5,905 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
import java.util.Arrays; import java.util.Scanner; public class Main { public static void main(String[] args) { new Main().run(); } final long MOD = 1000000007; void run() { Scanner sc = new Scanner(System.in); long N = sc.nextLong(); long[][] mat = { { 1, 1, 0, 0 }, { 0, 1, 1, 2 }, { 0, 1, 0, 0 }, { 0, 1, 0, 1 } }; long[][] v = { { 0 }, { 1 }, { 0 }, { 0 } }; mat = pow(mat, N); mat = mul(mat, v); System.out.println(mat[0][0]); } long[][] pow(long[][] a, long n) { long[][] ret = new long[a.length][a.length]; for (int i = 0; i < a.length; ++i) ret[i][i] = 1; for (; n > 0; n >>= 1, a = mul(a, a)) { if (n % 2 == 1) ret = mul(a, ret); } return ret; } long[][] mul(long[][] a, long[][] b) { long[][] ret = new long[a.length][b[0].length]; for (int i = 0; i < a.length; ++i) { for (int j = 0; j < b[i].length; ++j) { for (int k = 0; k < a[i].length; ++k) { ret[i][j] = (ret[i][j] + a[i][k] * b[k][j]) % MOD; } } } return ret; } void tr(Object... objects) { System.out.println(Arrays.deepToString(objects)); } }