結果

問題 No.2817 Competition
ユーザー zeta7532zeta7532
提出日時 2024-07-18 02:26:17
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 769 ms / 2,000 ms
コード長 2,282 bytes
コンパイル時間 2,396 ms
コンパイル使用メモリ 215,992 KB
実行使用メモリ 18,500 KB
最終ジャッジ日時 2024-07-18 02:26:29
合計ジャッジ時間 11,517 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 769 ms
18,496 KB
testcase_04 AC 730 ms
18,496 KB
testcase_05 AC 746 ms
18,500 KB
testcase_06 AC 745 ms
18,480 KB
testcase_07 AC 728 ms
18,492 KB
testcase_08 AC 84 ms
6,940 KB
testcase_09 AC 179 ms
7,248 KB
testcase_10 AC 254 ms
9,776 KB
testcase_11 AC 437 ms
12,876 KB
testcase_12 AC 3 ms
6,944 KB
testcase_13 AC 187 ms
7,384 KB
testcase_14 AC 306 ms
10,836 KB
testcase_15 AC 146 ms
6,952 KB
testcase_16 AC 310 ms
10,616 KB
testcase_17 AC 212 ms
7,952 KB
testcase_18 AC 751 ms
18,500 KB
testcase_19 AC 2 ms
6,944 KB
testcase_20 AC 1 ms
6,944 KB
testcase_21 AC 2 ms
6,940 KB
testcase_22 AC 2 ms
6,944 KB
testcase_23 AC 2 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
using namespace std;
using ll = long long;
const ll mod = 998244353;
#define fi first
#define se second
#define rep(i,n) for(ll i=0;i<n;i++)
#define all(x) x.begin(),x.end()
#define faster ios::sync_with_stdio(false);cin.tie(nullptr)

const ll root = 3;

ll modpow(ll A,ll N){
    ll res=1;
    while(N){
        if(N&1) res=res*A%mod;
        A=A*A%mod;
        N>>=1;
    }
    return res;
}

ll modinv(ll A){return modpow(A,mod-2);}

void NTT(vector<ll> &A,const bool rev=false){
    vector<ll> B(A.size(),0);
    ll R=mod-1-(mod-1)/((ll)A.size());
    if(rev) R=(mod-1)/((ll)A.size());
    ll s=modpow(root,R);
    vector<ll> power((ll)A.size()/2+1,1);
    rep(i,(ll)A.size()/2) power[i+1]=(power[i]*s)%mod;
    ll cur=(ll)A.size()/2;
    for(ll i=1;i<(ll)A.size();i*=2){
        rep(j,cur){
            rep(k,i){
                ll p=A[k+i*j];
                ll q=A[k+i*j+(ll)A.size()/2];
                B[k+2*i*j]=(p+q)%mod;
                B[k+2*i*j+i]=(p+mod-q)*power[i*j]%mod;
            }
        }
        swap(A,B);
        cur/=2;
    }
    if(rev){
        s=modinv((ll)A.size());
        rep(i,(ll)A.size()) A[i]=(A[i]*s)%mod;
    }
}

vector<ll> convolution(vector<ll> A,vector<ll> B){
    ll size_=1;
    while(size_<=(ll)A.size()+(ll)B.size()-1) size_*=2;
    while(A.size()<size_) A.push_back(0);
    while(B.size()<size_) B.push_back(0);
    NTT(A);
    NTT(B);
    rep(i,size_) A[i]=(A[i]*B[i])%mod;
    NTT(A,true);
    return A;
}

vector<ll> mult(vector<ll> A,vector<ll> B,ll deg){
    vector<ll> ans=convolution(A,B);
    if(ans.size()<deg) ans.push_back(0);
    vector<ll> ANS(deg);
    rep(i,deg) ANS[i]=ans[i];
    return ANS;
}

int main() {
    ll N;
    cin >> N;
    deque<vector<ll>> que;
    rep(i,N){
        ll a;
        cin >> a;
        que.push_back({1,a%mod});
    }
    while(que.size()>=2){
        auto q1=que.front();
        que.pop_front();
        auto q2=que.front();
        que.pop_front();
        que.push_back(mult(q1,q2,q1.size()+q2.size()-1));
    }
    vector<ll> A=que.front();
    ll ans=0;
    rep(i,N){
        ans+=modpow(N-i,i)*A[N-i];
        ans%=mod;
    }
    cout << ans << endl;
    return 0;
}
0