結果

問題 No.658 テトラナッチ数列 Hard
ユーザー Coki628Coki628
提出日時 2020-05-20 12:04:33
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 588 ms / 2,000 ms
コード長 3,783 bytes
コンパイル時間 2,577 ms
コンパイル使用メモリ 208,980 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-04-09 22:54:27
合計ジャッジ時間 5,551 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 4 ms
6,944 KB
testcase_04 AC 205 ms
6,940 KB
testcase_05 AC 236 ms
6,940 KB
testcase_06 AC 301 ms
6,944 KB
testcase_07 AC 327 ms
6,944 KB
testcase_08 AC 384 ms
6,940 KB
testcase_09 AC 588 ms
6,940 KB
testcase_10 AC 582 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

typedef long long ll;
typedef long double ld;
typedef pair<ll, ll> pll;
typedef pair<ll, int> pli;
typedef pair<int, int> pii;
typedef pair<ll, ld> pld;
typedef pair<pii, int> ppiii;
typedef pair<pii, ll> ppiil;
typedef pair<pll, ll> pplll;
typedef pair<pli, int> pplii;
typedef vector<vector<ll>> vvl;
typedef vector<vector<int>> vvi;
typedef vector<vector<pll>> vvpll;
#define rep(i, a, b) for (ll i=(a); i<(b); i++)
#define rrep(i, a, b) for (ll i=(a); i>(b); i--)
#define pb push_back
#define tostr to_string
#define mkp make_pair
const ll INF = 1e18;
const ll MOD = 17;

void print(ld out) { cout << fixed << setprecision(12) << out << '\n'; }
template<typename T> void print(T out) { cout << out << '\n'; }
template<typename T1, typename T2> void print(pair<T1, T2> out) { cout << out.first << ' ' << out.second << '\n'; }
template<typename T> void print(vector<T> A) { rep(i, 0, A.size()) { cout << A[i]; cout << (i == A.size()-1 ? '\n' : ' '); } }
template<typename T> void print(set<T> S) { vector<T> A(S.begin(), S.end()); print(A); }

template<typename T> inline bool chmax(T &x, T y) { return (y > x) ? x = y, true : false; }
template<typename T> inline bool chmin(T &x, T y) { return (y < x) ? x = y, true : false; }

ll sum(vector<ll> A) { ll res = 0; for (ll a: A) res += a; return res; }
ll max(vector<ll> A) { ll res = -INF; for (ll a: A) chmax(res, a); return res; }
ll min(vector<ll> A) { ll res = INF; for (ll a: A) chmin(res, a); return res; }

ll toint(string s) { ll res = 0; for (char c : s) { res *= 10; res += (c - '0'); } return res; }
int toint(char c) { return c - '0'; }
char tochar(int i) { return '0' + i; }

inline ll pow(int x, ll n) { ll res = 1; rep(_, 0, n) res *= x; return res; }
inline ll pow(ll x, ll n, int mod) { ll res = 1; while (n > 0) { if (n & 1) { res = (res * x) % mod; } x = (x * x) % mod; n >>= 1; } return res; }

inline ll floor(ll a, ll b) { if (a < 0) { return (a-b+1) / b; } else { return a / b; } }
inline ll ceil(ll a, ll b) { if (a >= 0) { return (a+b-1) / b; } else { return a / b; } }

int popcount(ll S) { return __builtin_popcountll(S); }
ll gcd(ll a, ll b) { return __gcd(a, b); }

struct MatPow {

    vvl mat_dot(vvl A, vvl B) {
        ll n1 = A.size();
        ll n2 = A[0].size();
        ll m2 = B[0].size();
        vvl res(n1, vector<ll>(m2, 0));
        rep(i, 0, n1) {
            rep(j, 0, m2) {
                rep(k, 0, n2) {
                    res[i][j] = (res[i][j] + (A[i][k] * B[k][j]) % MOD) % MOD;
                }
            }
        }
        return res;
    }
    
    vvl mat_pow(vvl mat, ll k) {
        ll n = mat.size();
        vvl res(n, vector<ll>(n, 0));
        rep(i, 0, n) {
            res[i][i] = 1;
        }
        while (k > 0) {
            if (k & 1) {
                res = mat_dot(res, mat);
            }
            mat = mat_dot(mat, mat);
            k >>= 1;
        }
        return res;
    }

    vector<ll> solve(vvl mat, vvl init, ll K) {
        auto res = mat_pow(mat, K);
        res = mat_dot(res, init);
        ll n = mat.size();
        vector<ll> ret(n, 0);
        rep(i, 0, n) {
            ret[i] = res[i][0];
        }
    return ret;
    }
};

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);

    vvl mat(4, vector<ll>(4, 0));
    rep(i, 0, 4) {
        mat[0][i] = 1;
    }
    rep(i, 0, 3) {
        mat[i+1][i] = 1;
    }
    vvl init(4, vector<ll>(1, 0));
    init[0][0] = 1;

    ll Q;
    cin >> Q;

    MatPow mp;
    ll n;
    rep(i, 0, Q) {
        cin >> n;
        if (n < 4) {
            print(0);
        } else if (n == 4) {
            print(1);
        } else {
            auto res = mp.solve(mat, init, n-4);
            print(res[0]);
        }
    }
    return 0;
}
0