結果

問題 No.980 Fibonacci Convolution Hard
ユーザー sugarrrsugarrr
提出日時 2020-07-15 14:47:04
言語 C++17(clang)
(14.0.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 7,908 bytes
コンパイル時間 2,766 ms
コンパイル使用メモリ 137,388 KB
実行使用メモリ 14,768 KB
最終ジャッジ日時 2023-08-20 16:08:03
合計ジャッジ時間 9,021 ms
ジャッジサーバーID
(参考情報)
judge13 / judge12
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 TLE -
testcase_01 -- -
testcase_02 -- -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

//#include <bits/stdc++.h>
#include "bits/stdc++.h"
using namespace std;
typedef long long ll;

/*
#include "boost/multiprecision/cpp_int.hpp"
#include "boost/multiprecision/cpp_dec_float.hpp"
namespace mp = boost::multiprecision;
typedef mp::cpp_int LL;
typedef mp::number<mp::cpp_dec_float<1024>> DD;// 仮数部が1024ビットの浮動小数点数型(TLEしたら小さくする)
*/
 
typedef long double dd;
#define i_7 (ll)(1E9+7)
//#define i_7 998244353
#define i_5 i_7-2
ll mod(ll a){
    ll c=a%i_7;
    if(c>=0)return c;
    return c+i_7;
}
typedef pair<ll,ll> l_l;
typedef pair<dd,dd> d_d;
ll inf=(ll)1E16;
#define rep(i,l,r) for(ll i=l;i<=r;i++)
#define rrep(i,r,l) for(ll i=r;i>=l;i--)
#define pb push_back
ll max(ll a,ll b){if(a<b)return b;else return a;}
ll min(ll a,ll b){if(a>b)return b;else return a;}
void Max(ll &pos,ll val){pos=max(pos,val);}//Max(dp[n],dp[n-1]);
void Min(ll &pos,ll val){pos=min(pos,val);}
void Add(ll &pos,ll val){pos=mod(pos+val);}
dd EPS=1E-9;
#define fastio ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
#define fi first
#define se second
//#define endl "\n"  //インタラクティブで消す!!!!!!!!!!!!!!!!!!!!!
#define SORT(v) sort(v.begin(),v.end())
#define ERASE(v) v.erase(unique(v.begin(),v.end()),v.end())
#define POSL(v,x) (lower_bound(v.begin(),v.end(),x)-v.begin())
#define POSU(v,x) (upper_bound(v.begin(),v.end(),x)-v.begin())
template<class T,class S>
inline bool chmax(T &a, S b) {
    if(a < b) {
        a = b;
        return true;
    }
    return false;
}
template<class T,class S>
inline bool chmin(T &a, S b) {
    if(a > b) {
        a = b;
        return true;
    }
    return false;
}

void mod_print(ll k){
    ll P=50000;
    rep(y,1,P){
        ll x=mod(y*k);
        
        if(abs(x)<=P||x+P>=i_7){
            if(x+P>=i_7){
                x-=i_7;
            }
            cout<<x<<"/"<<y<<endl;
            //cout<<setprecision(5)<<(dd)x/(dd)y;
            return;
        }
    }
    cout<<"nun"<<endl;
}
#define all(c) c.begin(),c.end()
typedef vector<ll> vl;
//////////////////////////


//from KaedeTakagaki
//from https://judge.yosupo.jp/submission/7396
template<const ll md>
struct ntt{
    inline void add(ll &a, ll b) { a += b; if(a >= md) a -= md; }
    inline void sub(ll &a, ll b) { a -= b; if(a < 0) a += md; }
    inline ll add2(ll a, ll b) { a += b; if(a >= md) a -= md; return a;}
    inline ll sub2(ll a, ll b) { a -= b; if(a < 0) a += md; return a;}
    inline ll mul(ll a, ll b) { return (ll)((ll)a*b%md); }
    inline ll power(ll a, long long b) {
        ll res = 1;
        while (b > 0) {
            if (b & 1) res = mul(res, a);
            a = mul(a, a);
            b >>= 1;
        }
        return res;
    }
    inline ll inv(ll a) {
        a %= md;
        if (a < 0) a += md;
        ll b = md, u = 0, v = 1;
        while (a) {
            ll t = b / a;
            b -= t * a; swap(a, b);
            u -= t * v; swap(u, v);
        }
        assert(b == 1);
        if (u < 0) u += md;
        return u;
    }
    
     ll max_base, root;
    vector<ll> dw, idw;
    ntt() {
        ll tmp = md - 1;
        max_base = 0;
        while (tmp % 2 == 0) {
            tmp /= 2;
            max_base++;
        }
        root = 2;
        while (power(root, (md-1)>>1) == 1) root++;
        dw.resize(max_base); idw.resize(max_base);
        
        rep(i,0,max_base-1){
            sub(dw[i], power(root, (md-1) >> (i+2)));
            idw[i] = inv(dw[i]);
        }
    }
    void fft(vector<ll> &a, bool inv) {
        const ll n = a.size();
        assert((n & (n - 1)) == 0);
        assert(__builtin_ctz(n) <= max_base);
        if(!inv){
            for(ll m=n;m>>=1;){
                ll w = 1;
                for(ll s=0,k=0; s<n; s += 2*m){
                    for(ll i=s, j=s+m ; i < s+m; ++i, ++j) {
                        ll x = a[i], y = mul(a[j], w);
                        a[j] = (x>=y?x-y:x+md-y);
                        a[i] = (x+y>=md?x+y-md:x+y);
                    }
                    w = mul(w, dw[__builtin_ctz(++k)]);
                }
            }
        }
        else{
            for(ll m=1;m<n;m*=2){
                ll w = 1;
                for(ll s=0,k=0; s<n; s += 2*m){
                    for(ll i=s, j=s+m ; i < s+m; ++i, ++j) {
                        ll x = a[i], y = a[j];
                        a[j] = (x>=y?x-y:x+md-y);
                        a[j] = mul(a[j], w);
                        a[i] = (x+y>=md?x+y-md:x+y);
                    }
                    w = mul(w, idw[__builtin_ctz(++k)]);
                }
            }
        }
    }
    vector<ll> multiply(vector<ll> a, vector<ll> b, ll eq = 0) {
        ll need = a.size() + b.size() - 1;
        ll nbase = 0;
        while ((1 << nbase) < need) nbase++;
        ll sz = 1 << nbase;
        a.resize(sz);
        b.resize(sz);
        fft(a, 0);
        if (eq) b = a; else fft(b, 0);
        ll inv_sz = inv(sz);
        for (ll i = 0; i < sz; i++) {
            a[i] = mul(mul(a[i], b[i]), inv_sz);
        }
        fft(a, 1);
        a.resize(need);
        return a;
    }
    vector<ll> square(vector<ll> a) {
        return multiply(a, a, 1);
    }
};
/*
167772161; //= 2^25 * 5 + 1
469762049; //= 2^26 * 7 + 1
754974721; //= 2^24 * 45 + 1
1045430273; //= 2^20 * 997 + 1
1051721729; //= 2^20 * 1003 + 1
1053818881; //= 2^20 * 1005 + 1
*/

template<const ll md>
vector<ll> anyntt(vector<ll>&a, vector<ll>&b, ll eq = 0) {
    
    for(auto &x:a)if(x<0)x=mod(x);
    for(auto &x:b)if(x<0)x=mod(x);
    
    //今回は2^20以下だからこっちの方が速い (らしい) (:maroon_kansha:)
    const ll m1 = 167772161, m2 = 469762049, m3 = 754974721;
    ntt<m1>n1;
    ntt<m2>n2;
    ntt<m3>n3;
    ntt<md>nn;
    auto a1  = n1.multiply(a, b, eq);
    auto a2 = n2.multiply(a, b, eq);
    auto a3  = n3.multiply(a, b, eq);
    const ll n = a1.size();
    vector<ll>ret(n);
    vector<ll>m; m = {m1, m2, m3};
    vector<ll>r; r = {1, n2.inv(m1), n3.inv(n3.mul(m1, m2))};
    ll mm = nn.mul(m1, m2);
    
    rep(i,0,n-1){
        n2.add(a2[i], n2.sub2(m2, a1[i]));
        ll v1 = n2.mul(a2[i], r[1]);
        n3.add(a3[i], n3.sub2(m3, n3.add2(a1[i], n3.mul(m1, v1))));
        ll v2 = n3.mul(a3[i], r[2]);
        nn.add(a1[i], nn.add2(nn.mul(m1, v1), nn.mul(mm, v2)));
        ret[i] = a1[i];
    }
    return ret;
}

ll po(ll i,ll p){
    if(p==0)return 1;
    else{
        i=mod(i);
        if(p==1)return i;
        if(p%2==0)return po(mod(i*i),p/2);
        return mod(i*po(i,p-1));
    }
}
ll bunbo(ll n){
    return po(n,i_5);
}

//from maspy
// https://atcoder.jp/contests/aising2020/submissions/15155062
ll coef_of_generating_function(vector<ll>P,vector<ll>Q,ll n){
    //多項式P,Qについて、PをQで割ったときのx^nの係数をreturn
    //Pの次数はQより小さいことが必要。
    
    for(auto &x:P)if(x<0)x=mod(x);
    for(auto &x:Q)if(x<0)x=mod(x);
    
    //Q[0]を1にする。
    ll z=Q[0];
    ll bz=bunbo(z);
    for(auto &x:P)x=mod(x*bz);
    for(auto &x:Q)x=mod(x*bz);
    
    ll sp=P.size(),sq=Q.size();
    
    rep(zz,1,sq-sp-1)P.pb(0);
    
    while(n>0){
        vector<ll>Q1(Q.size());
        rep(i,0,Q.size()-1){
            if(i%2==0){
                Q1[i]=Q[i];
            }else{
                Q1[i]=mod(-Q[i]);
            }
        }
        auto np=anyntt<(ll)i_7>(P,Q1);
        auto nq=anyntt<(ll)i_7>(Q,Q1);
        P.clear();Q.clear();
        for(ll i=n&1;i<=np.size()-1;i+=2)P.pb(np[i]);
        for(ll i=0;  i<=nq.size()-1;i+=2)Q.pb(nq[i]);
        n/=2;
    }
    return P[0];
}
///////////////////////



int main(){fastio
    ll p;cin>>p;
    vl a={0,0,1};
    vl b={1,-p,-1};
    b=anyntt<i_7>(b,b);
    ll que;cin>>que;
    while(que--){
        ll q;cin>>q;q-=2;
        cout<<coef_of_generating_function(a,b,q)<<endl;
    }
    
    return 0;
}
0