結果

問題 No.526 フィボナッチ数列の第N項をMで割った余りを求める
ユーザー zawakasuzawakasu
提出日時 2022-09-28 22:53:44
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,045 bytes
コンパイル時間 2,188 ms
コンパイル使用メモリ 210,096 KB
実行使用メモリ 4,504 KB
最終ジャッジ日時 2023-08-24 04:37:18
合計ジャッジ時間 2,952 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using i32 = int;
using i64 = long long;

vector<vector<i64>> mul(vector<vector<i64>>& as, vector<vector<i64>>& bs, i32 m) {
    assert(as[0].size() == bs.size());
    vector res(as.size(), vector(bs[0].size(), 0LL));
    for (size_t i = 0 ; i < as.size() ; i++) {
        for (size_t j = 0 ; j < bs[0].size() ; j++) {
            for (size_t k = 0 ; k < as[0].size() ; k++) {
                res[i][j] += (as[i][k] * bs[k][j]) % m;
                res[i][j] %= m;
            }
        }
    }
    return res;
}

int main() {
    i32 n; cin >> n;
    i32 m; cin >> m;
    n--;
    const i32 lg = 32;
    vector dob(lg, vector(2, vector(2, 0LL)));
    dob[0] = { { 1LL, 1LL }, { 1LL, 0LL } };
    for (i32 i = 0 ; i + 1 < lg ; i++) {
        dob[i + 1] = mul(dob[i], dob[i], m);
    }
    vector<vector<i64>> ans = { { 1LL, 0LL }, { 0LL, 1LL } };
    for (i32 i = 0 ; i < 10 ; i++) {
        if (n & (1 << i)) {
            ans = mul(ans, dob[i], m);
        }
    }
    cout << ans[1][0] << endl;
}
0