結果

問題 No.658 テトラナッチ数列 Hard
コンテスト
ユーザー kurimupy
提出日時 2020-06-15 23:27:38
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 625 ms / 2,000 ms
コード長 1,464 bytes
コンパイル時間 1,832 ms
コンパイル使用メモリ 177,132 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-03 11:38:06
合計ジャッジ時間 5,278 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

const long long mod = 17;
vector<vector<long long>> matprod(vector<vector<long long>> m1, vector<vector<long long>> m2, int n, int l, int m){
    vector<vector<long long>> y(n, vector<long long>(m));
    for (int i=0;i<n;i++){
        for (int j=0;j<m;j++){
            for (int k=0;k<l;k++){
                y[i][j] += m1[i][k] * m2[k][j] % mod;
                y[i][j] %= mod;
            }
        }
    }
    return y;
}
vector<vector<long long>> matpow(vector<vector<long long>> m1, int n, long long m){
    vector<vector<long long>> y(n, vector<long long>(n));
    vector<vector<long long>> bas = m1;
    for (int i=0;i<n;i++){
        y[i][i] = 1;
    }
    long long tmp = m;
    while (tmp != 0){
        if (tmp % 2 == 1){
            y = matprod(y, bas, n, n, n);
        }
        bas = matprod(bas, bas, n, n, n);
        tmp /= 2;
    }
    return y;
}
int main(void){
    // Your code here!
    int Q;
    cin >> Q;
    vector<long long> vec = {0, 0, 0, 1};
    vector<vector<long long>> mat = {
        {0, 1, 0, 0},
        {0, 0, 1, 0},
        {0, 0, 0, 1},
        {1, 1, 1, 1}
    };
    long long n;
    long long S;
    vector<vector<long long>> M;
    for (int _=0;_<Q;_++){
        cin >> n;
        M = matpow(mat, 4, n-1);
        S = 0;
        for (int i=0;i<4;i++){
            S += vec[i] * M[0][i] % mod;
            S %= mod;
        }
        cout << S << endl;
    }
    return 0;
}
0