結果

問題 No.658 テトラナッチ数列 Hard
コンテスト
ユーザー ngtkana
提出日時 2020-04-04 12:43:21
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 54 ms / 2,000 ms
コード長 1,320 bytes
コンパイル時間 1,980 ms
コンパイル使用メモリ 201,712 KB
最終ジャッジ日時 2025-01-09 14:01:06
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using lint=long long;
using real=long double;
int main(){
    std::cin.tie(nullptr);std::ios_base::sync_with_stdio(false);
    std::cout.setf(std::ios_base::fixed);std::cout.precision(15);
    lint mod=17;
    std::vector<lint>q={1,-1,-1,-1,-1},qinv={1,1,1,1},pa={1,-1,-1,-1};
    auto mul=[&,qdinv=-1](auto&&a,auto&&b){
        std::vector<lint>c(7);
        for(lint i=0;i<4;i++){
        for(lint j=0;j<4;j++){
            c.at(i+j)+=a.at(i)*b.at(j);
        }}
        while(4u<c.size()){
            lint x=-c.back()*qdinv;
            for(lint i=0;i<(lint)q.size();i++){
                c.at(c.size()-5+i)+=x*q.at(i);
            }
            assert(c.back()%mod==0);
            c.pop_back();
        }
        for(lint&x:c){
            x%=mod;
            if(x<0)x+=mod;
        }
        return c;
    };
    lint lg=63;
    std::vector<std::vector<lint>>table(lg);
    table.at(0)=qinv;
    for(lint i=1;i<lg;i++){
        table.at(i)=mul(table.at(i-1),table.at(i-1));
    }
    auto apow=[&](lint x){
        std::vector<lint>ans={1,0,0,0};
        for(lint i=0;i<lg;i++){
            if(x>>i&1)ans=mul(ans,table.at(i));
        };
        return ans;
    };
    lint t;std::cin>>t;
    while(t--){
        lint n;std::cin>>n;
        std::cout<<mul(apow(n),pa).at(0)<<'\n';
    }
}
0