結果
問題 |
No.720 行列のできるフィボナッチ数列道場 (2)
|
ユーザー |
👑 ![]() |
提出日時 | 2020-10-11 18:05:13 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 859 bytes |
コンパイル時間 | 2,538 ms |
コンパイル使用メモリ | 166,916 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-07-20 17:32:14 |
合計ジャッジ時間 | 2,965 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
#include<bits/stdc++.h> using namespace std; using LL = long long; using ULL = unsigned long long; #define rep(i,n) for(int i=0; i<(n); i++) const ULL M = 1000000007; const int matrix_size = 3; struct Matrix { ULL X[matrix_size][matrix_size] = {}; static Matrix id() { Matrix res; rep(i, matrix_size) res.X[i][i] = 1; return res; } }; Matrix operator*(Matrix l, Matrix r) { Matrix res; rep(i, matrix_size) rep(j, matrix_size) rep(k, matrix_size) res.X[i][j] = (res.X[i][j] + l.X[i][k] * r.X[k][j]) % M; return res; } Matrix operator^(Matrix l, ULL i) { if (i == 0) return Matrix::id(); auto res = (l * l) ^ (i / 2); if (i % 2 == 1) res = res * l; return res; } int main() { ULL N, I; cin >> N >> I; Matrix X = { { {1,1,0},{1,0,0},{0,0,1} } }; X = X ^ I; X.X[0][2] = 1; X = X ^ (N + 1); cout << X.X[1][2] << endl; return 0; }