結果
問題 |
No.526 フィボナッチ数列の第N項をMで割った余りを求める
|
ユーザー |
|
提出日時 | 2023-10-03 19:30:45 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 2,709 bytes |
コンパイル時間 | 4,100 ms |
コンパイル使用メモリ | 237,188 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-07-26 14:09:32 |
合計ジャッジ時間 | 4,565 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 12 |
ソースコード
#include <bits/stdc++.h> using namespace std; #include <atcoder/all> using namespace atcoder; typedef long long ll; typedef unsigned long long ull; const int MAX = 1e9; const int MIN = -1*1e9; const ll MAXLL = 1e18; const ll MINLL = -1*1e18; //const ll MOD = 998244353; //const ll MOD = 1000000007; template<class T> struct Matrix { vector<vector<T> > val; Matrix(int n = 1, int m = 1, T v = 0) : val(n, vector<T>(m, v)) {} void init(int n, int m, T v = 0) {val.assign(n, vector<T>(m, v));} void resize(int n, int m) { val.resize(n); for (int i = 0; i < n; ++i) val[i].resize(m); } Matrix<T>& operator = (const Matrix<T> &A) { val = A.val; return *this; } size_t size() const {return val.size();} vector<T>& operator [] (int i) {return val[i];} const vector<T>& operator [] (int i) const {return val[i];} friend ostream& operator << (ostream& s, const Matrix<T>& M) { s << endl; for (int i = 0; i < (int)M.size(); ++i) s << M[i] << endl; return s; } }; template<class T> Matrix<T> operator * (const Matrix<T> &A, const Matrix<T> &B) { Matrix<T> R(A.size(), B[0].size()); for (int i = 0; i < A.size(); ++i) for (int j = 0; j < B[0].size(); ++j) for (int k = 0; k < B.size(); ++k) R[i][j] += A[i][k] * B[k][j]; return R; } template<class T> Matrix<T> pow(const Matrix<T> &A, long long n) { Matrix<T> R(A.size(), A.size()); auto B = A; for (int i = 0; i < A.size(); ++i) R[i][i] = 1; while (n > 0) { if (n & 1) R = R * B; B = B * B; n >>= 1; } return R; } template<class T> vector<T> operator * (const Matrix<T> &A, const vector<T> &B) { vector<T> v(A.size()); for (int i = 0; i < A.size(); ++i) for (int k = 0; k < B.size(); ++k) v[i] += A[i][k] * B[k]; return v; } template<class T> Matrix<T> operator + (const Matrix<T> &A, const Matrix<T> &B) { Matrix<T> R(A.size(), A[0].size()); for (int i = 0; i < A.size(); ++i) for (int j = 0; j < A[0].size(); ++j) R[i][j] = A[i][j] + B[i][j]; return R; } template<class T> Matrix<T> operator - (const Matrix<T> &A, const Matrix<T> &B) { Matrix<T> R(A.size(), A[0].size()); for (int i = 0; i < A.size(); ++i) for (int j = 0; j < A[0].size(); ++j) R[i][j] = A[i][j] - B[i][j]; return R; } int main() { ll N,M; cin >> N >> M; using mint = modint; mint::set_mod(M); Matrix<mint> A(2,2,0); Matrix<mint> B(1,2,0); A[0][0] = 1; A[0][1] = 1; A[1][0] = 1; A[1][1] = 0; A = pow(A,N-2); cout << A[0][0].val() << endl; return 0; }