結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 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 2 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 2 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(std::vector<std::vector<int64_t>> A,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 pow_mod(int64_t a,int64_t n,int64_t mod)
  {
    assert(0<=n);
    int64_t res=1;
    while(0<n)
    {
      if(0<(n&1))
      {
        res=res*a%mod;
      }
      a=a*a%mod;
      n=n>>1;
    }
    return res;
  }

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