結果

問題 No.526 フィボナッチ数列の第N項をMで割った余りを求める
ユーザー ytqm3
提出日時 2021-05-01 17:38:36
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,036 bytes
コンパイル時間 1,969 ms
コンパイル使用メモリ 203,260 KB
最終ジャッジ日時 2025-01-21 05:25:12
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>

std::vector<std::vector<int64_t>> matrix_product(const std::vector<std::vector<int64_t>> &A,const std::vector<std::vector<int64_t>> &B,const int64_t mod)
{
  int64_t AH=A.size(),AW=A[0].size(),BH=B.size(),BW=B[0].size();
  assert(AW==BH);
  std::vector<std::vector<int64_t>> C(AH,std::vector<int64_t> (BW));
  for(int i=0;i<AH;i++)
  {
    for(int j=0;j<BW;j++)
    {
      for(int k=0;k<AH;k++)
      {
        C[i][j]+=A[i][k]*B[k][j];
      }
      C[i][j]%=mod;
    }
  }
  return C;
}

int64_t fibonacci(int64_t N,const int64_t mod)
{
  if(N==0)
  {
    return 0;
  }
  N--;
  std::vector<std::vector<int64_t>> F(2,std::vector<int64_t> (1)),A(2,std::vector<int64_t> (2)),B;
  F[0][0]=1,F[1][0]=0;
  A[0][0]=1,A[0][1]=1,A[1][0]=1,A[1][1]=0;
  B=A;
  while(0<N)
  {
    if((N&1))
    {
      A=matrix_product(A,B,mod);
    }
    B=matrix_product(B,B,mod);
    N>>=1;
  }
  return matrix_product(A,F,mod)[1][0];
}

int main()
{
  int64_t N,M;
  std::cin>>N>>M;
  N--;
  std::cout<<fibonacci(N,M)<<std::endl;
}
0