結果
問題 | No.754 畳み込みの和 |
ユーザー | itsutei |
提出日時 | 2021-08-26 18:33:08 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 293 ms / 5,000 ms |
コード長 | 1,801 bytes |
コンパイル時間 | 4,035 ms |
コンパイル使用メモリ | 246,408 KB |
実行使用メモリ | 14,680 KB |
最終ジャッジ日時 | 2024-04-29 09:55:18 |
合計ジャッジ時間 | 5,714 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 293 ms
14,680 KB |
testcase_01 | AC | 292 ms
14,672 KB |
testcase_02 | AC | 292 ms
14,676 KB |
ソースコード
#include<bits/stdc++.h> #include<atcoder/modint> #include<atcoder/convolution> using namespace atcoder; using mint = modint1000000007; using namespace std; using ll = long long; using P = pair<ll,ll>; ll exgcd(ll a,ll b,ll &x,ll &y){ x=0, y=1; ll u=1, v=0; while(a){ ll t{b/a}; swap(b-=t*a, a); swap(x-=t*u, u); swap(y-=t*v, v); } if(b<0) b*=-1, x*=-1, y*=-1; return b; } ll inv(ll a, ll MOD){ ll x{0},y{1}; exgcd(a,MOD,x,y); return (x%MOD+MOD)%MOD; } ll garner(vector<ll>&b,vector<ll>&m){ int sz = m.size(); vector<ll> coef(sz,1), cons(sz,0); for(int k=0; k<(int)b.size(); k++){ ll t = (b[k]-cons[k]); t *= inv(coef[k],m[k]); t = ( (t%m[k]) + m[k] ) % m[k]; for(int i=k+1; i<sz; i++){ (cons[i] += t*coef[i] ) %= m[i]; (coef[i] *= m[k]) %= m[i]; } } return cons.back(); } vector<mint> mod_conv(vector<mint>a,vector<mint>b){ int MOD = mint::mod(); const int nhg2 = 167772161; const int nhg3 = 469762049; const int nhg4 = 1224736769; vector<ll>c(a.size()),d(b.size()); for(int i=0; i<(int)a.size(); i++){ c[i] = a[i].val(); } for(int i=0; i<(int)b.size(); i++){ d[i] = b[i].val(); } auto x = convolution<nhg2>(c,d); auto y = convolution<nhg3>(c,d); auto z = convolution<nhg4>(c,d); int n = x.size(); vector<mint> ret(n); vector<ll> f {nhg2,nhg3,nhg4,MOD}; for(int i=0; i<n; i++){ vector<ll> e {x[i],y[i],z[i]}; ret[i] = garner(e,f); } return ret; } int main(){ int n; cin>>n; vector<mint>a(n+1),b(n+1); for(int i=0; i<=n; i++){ int t; cin>>t; a[i] = t; } for(int i=0; i<=n; i++){ int t; cin>>t; b[i] = t; } auto c = mod_conv(a,b); mint ans{}; for(int i=0; i<=n; i++){ ans += c[i]; } cout<<ans.val()<<endl; }