結果

問題 No.658 テトラナッチ数列 Hard
コンテスト
ユーザー wk1080id
提出日時 2019-06-01 01:40:58
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 985 ms / 2,000 ms
コード長 2,063 bytes
コンパイル時間 950 ms
コンパイル使用メモリ 90,880 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-17 19:52:08
合計ジャッジ時間 5,755 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cctype>
#include<math.h>
#include<string>
#include<string.h>
#include<stack>
#include<queue>
#include<vector>
#include<utility>
#include<set>
#include<map>
#include<stdlib.h>
#include<iomanip>

using namespace std;

#define ll long long
#define ld long double
#define EPS 0.0000000001
#define INF 1e9
#define LINF (ll)INF*INF
#define MOD 1000000007
#define rep(i,n) for(int i=0;i<(n);i++)
#define loop(i,a,n) for(int i=a;i<(n);i++)
#define all(in) in.begin(),in.end()
#define shosu(x) fixed<<setprecision(x)

#define int ll //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

typedef vector<int> vi;
typedef vector<string> vs;
typedef pair<int,int> pii;
typedef vector<pii> vp;

int gcd(int a, int b){
    if(b==0) return a;
    return gcd(b,a%b);
}
int lcm(int a, int b){
    return a/gcd(a,b)*b;
}

typedef vector<vi> vvi;

void printMatrix(vvi a){
    rep(i,a.size()){
        rep(j,a[0].size())cout << a[i][j] << " " ;
        cout << endl;
    }
}

vvi product(vvi a, vvi b, int mod){//ret = a * b % mod
    vvi ret(a.size(), vi(b[0].size(),0));
    rep(i,a.size())rep(j,b[0].size()){
        rep(k,a[0].size()) (ret[i][j] += a[i][k]*b[k][j]) %= mod;
    }
    return ret;
}


vvi powMatrix(vvi a, int t, int mod){// A^t % mod
    int n = a.size();

    vvi ret(n, vi(n,0));
    rep(i,n)ret[i][i] = 1; // 単位行列

    while(t > 0){ //繰り返し自乗法
        if(t&1) ret = product(ret, a, mod);
        a = product(a,a, mod);
        t = t >> 1;
    }
    return ret;
}


signed main(void) {
    vvi s(4,vi(1,0));
    s[0][0] = 1;
    vvi t(4, vi(4,0));
    t[0][0] = t[0][1] = t[0][2] = t[0][3] = t[1][0] = t[2][1] = t[3][2] = 1;
    int q;
    cin >> q;
    while(q--){
        int n;
        cin >> n;
        if(n < 5){
            cout << s[4-n][0] << endl;
        }else{
            vvi a = t;
            a = powMatrix(a,n-4,17);
            //printMatrix(a);
            vvi ans = product(a,s,17);
            cout << ans[0][0] << endl;
        }
    }
}
0