結果

問題 No.754 畳み込みの和
ユーザー ttttan2ttttan2
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

#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;
}



0