結果
問題 |
No.891 隣接3項間の漸化式
|
ユーザー |
![]() |
提出日時 | 2021-11-03 14:24:14 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 1,267 bytes |
コンパイル時間 | 803 ms |
コンパイル使用メモリ | 74,240 KB |
実行使用メモリ | 6,820 KB |
最終ジャッジ日時 | 2024-10-13 03:50:33 |
合計ジャッジ時間 | 1,715 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 39 |
ソースコード
#include <iostream> #include <vector> using namespace std; typedef vector<vector<int64_t>> Matrix; // 行列の乗算(mod) Matrix ModMultiMatrix(const Matrix &a, const Matrix &b, int64_t mod) { Matrix c(a.size(), vector<int64_t>(b.at(0).size())); for(int i = 0; i < a.size(); i++) { for(int j = 0; j < b.size(); j++) { for(int k = 0; k < b.at(0).size(); k++) { c.at(i).at(k) += a.at(i).at(j) * b.at(j).at(k); c.at(i).at(k) %= mod; } } } return c; } // 行列の累乗(mod) Matrix ModPowerMatrix(Matrix a, int64_t n, int64_t mod) { Matrix b(a.size(), vector<int64_t>(a.size())); for(int i = 0; i < a.size(); i++) b.at(i).at(i) = 1; while(n > 0) { if(n & 1) b = ModMultiMatrix(b, a, mod); a = ModMultiMatrix(a, a, mod); n >>= 1; } return b; } int main() { int64_t A, B, N; cin >> A >> B >> N; if(N == 0) { cout << 0 << endl; return 0; } int64_t mod = 1000000007; Matrix mat(2, vector<int64_t>(2)); mat.at(0).at(0) = A; mat.at(0).at(1) = B; mat.at(1).at(0) = 1; mat = ModPowerMatrix(mat, N - 1, mod); Matrix F(2, vector<int64_t>(1)); F.at(0).at(0) = 1; Matrix ans = ModMultiMatrix(mat, F, mod); cout << ans.at(0).at(0) << endl; return 0; }