結果
問題 | No.754 畳み込みの和 |
ユーザー | ttttan2 |
提出日時 | 2020-02-01 19:00:24 |
言語 | C++11 (gcc 11.4.0) |
結果 |
AC
|
実行時間 | 292 ms / 5,000 ms |
コード長 | 3,671 bytes |
コンパイル時間 | 1,730 ms |
コンパイル使用メモリ | 172,808 KB |
実行使用メモリ | 11,168 KB |
最終ジャッジ日時 | 2024-09-18 20:10:47 |
合計ジャッジ時間 | 3,907 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 288 ms
11,164 KB |
testcase_01 | AC | 292 ms
11,164 KB |
testcase_02 | AC | 292 ms
11,168 KB |
ソースコード
#include<bits/stdc++.h> //ios::sync_with_stdio(false);cin.tie(0);cout.tie(0); using namespace std; typedef long long ll; typedef unsigned long long ull; typedef unsigned int ui; typedef pair<int,int> pii; typedef pair<pii,int> ppii; typedef pair<int,pii> pipi; typedef pair<ll,ll> pll; typedef pair<pll,ll> ppll; typedef pair<ll,pll> plpl; typedef tuple<ll,ll,ll> tl; ll mod=1000000007; ll mod2=998244353; ll mod3=1000003; ll mod4=998244853; ll inf=1LL<<60; long double pi=3.14159265358979323846L; double eps=1e-12; #define rep(i,m,n) for(int i=m;i<n;i++) #define rrep(i,n,m) for(int i=n;i>=m;i--) #define srep(itr,st) for(auto itr=st.begin();itr!=st.end();itr++) int dh[4]={1,-1,0,0}; int dw[4]={0,0,1,-1}; int ddh[8]={-1,-1,-1,0,0,1,1,1}; int ddw[8]={-1,0,1,-1,1,-1,0,1}; ll gcd(ll a,ll b){ if(a<b)swap(a,b); if(b==0)return a; if(a%b==0)return b; return gcd(b,a%b); } ll Pow(ll n,ll k){ ll ret=1; ll now=n; while(k>0){ if(k&1)ret*=now; now*=now; k/=2; } return ret; } template<int md,int proot> struct NTT{ vector<long long>pp,invpp;//memoize proot^(mod-1>>i) and inv long long power(long long a,int b) { long long ret=1; while(b) { if(b&1)ret=ret*a%md; a=a*a%md; b>>=1; } return ret; } void dft(vector<int>&A,bool sign,int id) { if(id==0)return; int N=1<<id-1; vector<int>F(N),G(N); for(int i=0;i<N;i++) { F[i]=A[i<<1]; G[i]=A[i<<1|1]; } dft(F,sign,id-1); dft(G,sign,id-1); long long z=(sign?invpp:pp)[id],p=1; for(int i=0;i<N;i++) { A[i]=(F[i]+p*G[i])%md; A[i+N]=(F[i]-p*G[i])%md; if(A[i+N]<0)A[i+N]+=md; (p*=z)%=md; } } vector<int>multiply(vector<int>A,vector<int>B) { if(A.empty()||B.empty()) { return(vector<int>){}; } int N=1,sz=0; vector<int>ret(A.size()+B.size()-1); while(N<ret.size())N<<=1,sz+=1; pp.resize(sz+1); invpp.resize(sz+1); pp[sz]=power(proot,md-1>>sz); invpp[sz]=power(pp[sz],md-2); for(int i=sz-1;i>0;i-=1) { pp[i]=pp[i+1]*pp[i+1]%md; invpp[i]=invpp[i+1]*invpp[i+1]%md; } A.resize(N); B.resize(N); dft(A,false,sz); dft(B,false,sz); for(int i=0;i<N;i++)A[i]=(long long)A[i]*B[i]%md; dft(A,true,sz); long long invN=power(N,md-2); for(int i=0;i<ret.size();i++)ret[i]=invN*A[i]%md; return ret; } }; vector<int>multiply(vector<int>A,vector<int>B,const int md) { for(int&a:A) { a%=md; if(a<0)a+=md; } for(int&b:B) { b%=md; if(b<0)b+=md; } vector<int>C1=NTT<998244353,3>().multiply(A,B); vector<int>C2=NTT<469762049,3>().multiply(A,B); vector<int>C3=NTT<167772161,3>().multiply(A,B); vector<int>C(C1.size()); for(int i=0;i<C.size();i++) { long long v1=C1[i]; long long v2=(C2[i]-v1)*208783132%469762049; if(v2<0)v2+=469762049; long long v3=(C3[i]-v1-v2*998244353)%167772161*29562547%167772161; if(v3<0)v3+=167772161; C[i]=(v1+(v2+v3*469762049)%md*998244353)%md; } return C; } int main(){ ios::sync_with_stdio(false);cin.tie(0);cout.tie(0); int n;cin>>n; vector<int> a(n+1),b(n+1); rep(i,0,n+1)cin>>a[i]; rep(i,0,n+1)cin>>b[i]; vector<int>c=multiply(a,b,(int)mod); ll ans=0; rep(i,0,n+1)(ans+=c[i])%=mod; cout<<ans<<endl; }