結果
問題 | No.718 行列のできるフィボナッチ数列道場 (1) |
ユーザー |
👑 |
提出日時 | 2022-10-22 00:03:21 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 1,884 bytes |
コンパイル時間 | 1,062 ms |
コンパイル使用メモリ | 82,000 KB |
最終ジャッジ日時 | 2025-02-08 10:40:17 |
ジャッジサーバーID (参考情報) |
judge1 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
#include <iostream> #include <vector> using namespace std; const long long MOD = 1000000007; template <typename T> struct mat{ vector<vector<T>> x; int h,w; mat():x(vector<vector<T>>()){} mat(int h,int w):x(vector<vector<T>>(h,vector<T>(w))),h(h),w(w){} mat(int h,int w, T c):x(vector<vector<T>>(h,vector<T>(w,c))),h(h),w(w){} mat(vector<vector<T>> A):x(A),h(A.size()),w(A[0].size()){ } vector<T>& operator[](int i){return x[i];} mat& operator*=(mat& y){ mat<T> ret(h,y.w,0); if(w != y.h){ for(int i = 0; h > i; i++){ for(int j = 0; y.w > j; j++){ ret[i][j] = -1; } } }else{ for(int i = 0; h > i; i++){ for(int j = 0; y.w > j; j++){ for(int k = 0; w > k; k++){ ret[i][j] = (ret[i][j] + x[i][k]*y[k][j])%MOD; } } } } for(int i = 0; h > i; i++){ x[i].resize(y.w); } w = y.w; for(int i = 0; h > i; i++){ for(int j = 0; y.w > j; j++){ x[i][j] = ret[i][j]; } } return *this; } mat operator*(mat& y){return mat(*this) *= y;} mat pow(long long n){//正方行列のみ mat<T> res(h,w); mat<T> ret(h,w,0); mat<T> a(h,w); for(int i = 0; h > i; i++){ ret[i][i] = 1; } for(int i = 0; h > i; i++){ for(int j = 0; w > j; j++){ a[i][j] = (*this)[i][j]; } } while(n > 0){ if(n & 1){ ret *= a; } a *= a; n/=2; } for(int i = 0; h > i; i++){ for(int j = 0; w > j; j++){ res[i][j] = ret[i][j]; } } return res; } }; int main(){ vector<vector<long long>> x{{1,2,1,1},{1,0,0,0},{0,2,1,1},{0,0,1,0}}; vector<vector<long long>> y{{1},{0},{1},{0}}; mat<long long> A(x); mat<long long> B(y); long long n;cin>>n; auto C = A.pow(n-1)*B; cout << C[0][0] << endl; }