結果

問題 No.891 隣接3項間の漸化式
ユーザー oevloevl
提出日時 2019-09-20 23:02:48
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 830 bytes
コンパイル時間 1,754 ms
コンパイル使用メモリ 175,324 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-14 19:28:58
合計ジャッジ時間 2,955 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define rep(i, m, n) for (int i = m; i < n; ++i)
const long long MOD = 1000000007;

using vec = vector<long long>;
using Mat = vector<vec>;

Mat Mul(Mat A, Mat B) {
    Mat C(A.size(), vec(B[0].size()));
    rep(i, 0, A.size()) rep(k, 0, B.size()) rep(j, 0, B[0].size()) {
        C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % MOD;
    }
    return C;
}

Mat Pow(Mat A, long long n) {
    int SZ = A.size();
    Mat B(SZ, vec(SZ));
    rep(i, 0, SZ) B[i][i] = 1;
    while(n > 0) {
        if(n & 1) B = Mul(B, A);
        A = Mul(A, A);
        n >>= 1;
    }
    return B;
}

int main() {
    long long a, b, n;
    cin >> a >> b >> n;
    Mat A(2, vec(2));
    A[0][0] = a; A[0][1] = b;
    A[1][0] = 1LL; A[1][1] = 0LL;
    A = Pow(A, n);
    cout << A[1][0] << endl;
    return 0;
}
0