結果

問題 No.718 行列のできるフィボナッチ数列道場 (1)
ユーザー tetsu
提出日時 2018-07-27 22:50:14
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 657 bytes
コンパイル時間 1,731 ms
コンパイル使用メモリ 175,056 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-05 03:10:30
合計ジャッジ時間 2,524 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define rep(i,n) for(int (i)=0;(i)<(n);(i)++)

using mat = vector<vector<ll>>;

const ll p = 1e9+7;

mat mul(mat A, mat B) {
  mat ret = {{0,0},{0,0}};
  rep(i,2) {
    rep(j,2) {
      rep(k,2) {
	ret[i][j]+=A[i][k]*B[k][j];
	ret[i][j]%=p;
      }
    }
  }
  return ret;
}

mat matpow(mat &A, ll n) {
  if(n==0) return {{1,0},{0,1}};
  if(n%2==0) {
    mat AA = matpow(A,n/2);
    return mul(AA,AA);
  }
  return mul(A,matpow(A,n-1));
}

int main() {
  ll n;cin>>n;
  mat A = {{1,1},{1,0}};
  mat C = matpow(A,n-1);
  mat D = mul(C,A);
  cout<< (C[0][0]*D[0][0])%p << endl;
  return 0;
}
0