結果

問題 No.526 フィボナッチ数列の第N項をMで割った余りを求める
コンテスト
ユーザー ponta123
提出日時 2022-07-27 13:50:50
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 931 bytes
コンパイル時間 1,743 ms
コンパイル使用メモリ 176,076 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-16 19:20:54
合計ジャッジ時間 2,538 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define mat vector<vector<long long>>

mat matmul(mat &a, mat &b, long long M) {
    mat res(a.size(), (vector<long long>)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++) {
                (res[i][j] += a[i][k] * b[k][j]) %= M;
            }
        }
    }
    return res;
}

mat matpow(mat &a, long long n, long long M) {
    mat res(a.size(), (vector<long long>)a.size());
    for (int i = 0; i < a.size(); i++) {
        res[i][i] = 1;
    }
    while (n > 0) {
        if (n & 1) res = matmul(a, res, M);
        a = matmul(a, a, M);
        n >>= 1;
    }
    return res;
}

int main() {
    long long N, M;
    cin >> N >> M;
    mat res(2, (vector<long long>)2);
    res = {{1, 1}, {1, 0}};
    res = matpow(res, N - 3, M);
    cout << (res[0][0] + res[0][1]) % M << endl;
}
0