結果

問題 No.526 フィボナッチ数列の第N項をMで割った余りを求める
ユーザー ytqm3ytqm3
提出日時 2021-05-01 17:38:36
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,036 bytes
コンパイル時間 2,020 ms
コンパイル使用メモリ 210,040 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-20 01:41:36
合計ジャッジ時間 2,608 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 1 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 1 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 1 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

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