結果

問題 No.526 フィボナッチ数列の第N項をMで割った余りを求める
コンテスト
ユーザー mahala_coder
提出日時 2020-03-16 21:53:01
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 839 bytes
コンパイル時間 1,730 ms
コンパイル使用メモリ 172,224 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-28 03:03:31
合計ジャッジ時間 2,128 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other AC * 1 WA * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef vector<ll> vec;
typedef vector<vec> mat;
ll mod=1e9+7;

// size of matrix
ll m;
vec matmul(vec &dp, mat &mt) {
  vec ret(m);
  for(ll i=0;i<m;i++) {
    for(ll j=0;j<m;j++) {
      ret[i]=(ret[i]+mt[i][j]*dp[j])%mod;
    }
  }
  return ret;
}

mat update(mat &mt) {
  mat ret(m,vec(m));
  for(ll i=0;i<m;i++) {
    for(ll j=0;j<m;j++) {
      for(ll k=0;k<m;k++) {
        ret[i][j]=(ret[i][j]+mt[i][k]*mt[k][j])%mod;
      }
    }
  }
  return ret;
}

void matpow(vec &dp,mat &mt, ll k) {
  ll m=dp.size();
  while(k)  {
    if(k&1) dp=matmul(dp,mt);
    mt=update(mt);
    k/=2;
  }
}

int main() {
  m=2;
  vec dp(2);
  mat mt(2,vec(2));
  ll k;
  cin >> k >> mod;
  dp[0]=dp[1]=1;
  mt[0][0]=mt[0][1]=mt[1][0]=1;
  matpow(dp,mt,k+1);
  cout << dp[1] << endl;
}
0