結果

問題 No.526 フィボナッチ数列の第N項をMで割った余りを求める
コンテスト
ユーザー bonKotsu
提出日時 2022-02-04 15:51:30
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,217 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,434 ms
コンパイル使用メモリ 202,076 KB
最終ジャッジ日時 2025-01-27 18:34:50
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define rep(i, n) for (int i = 0; i < (n); i++)
#define P pair<int, int>
#define LP pair<ll, ll>
#define fi first
#define se second
#define pb push_back
#define eb emplace_back
#define all(s) s.begin(), s.end()
#define rall(s) s.rbegin(), s.rend()
typedef vector<ll> vl;
typedef vector<vl> mat;

template<class T>
void chmax(T& a, T b) { a = max(a, b); };
template<class T>
void chmin(T& a, T b) { a = min(a, b); };

ll m;
const mat e = {{1LL,0LL},{0LL,1LL}};



mat mul(mat &a, mat &b) {
  mat res(2,vl(2));
  rep(i,2)rep(j,2)rep(k,2) {
    res[i][j] += a[i][k]*b[k][j];
    res[i][j] %= m;
  }
  return res;
}

mat pow(mat &a, int n) {
  if (n == 1) return a;
  if (n == 0) return e;
  if (n%2==0) {
    mat t = pow(a,n/2);
    return mul(t,t);
  } else {
    mat t = pow(a,n/2);
    mat s = mul(t,t);
    return mul(a,s);
  }
}

vl mul(mat &a, vl &b) {
  vl res(2);
  rep(i,2)rep(k,2) {
    res[i] += a[i][k]*b[k];
    res[i] %= m;
  }
  return res;  
}

int main() {
  int n;
  cin >> n >> m;
  mat c = {{1LL,1LL},{1LL,0LL}};
  mat a = pow(c,n-1);
  vl v = {1LL,0LL};
  v = mul(a, v);
  ll ans = v[1];
  
  cout << ans << endl;

  
  return 0;
}
0