bzoj2179 FFT快速傅立叶

NTT。

#include<cstdio>
#include<algorithm>
using namespace std;
#define LL long long
const int p=998244353,maxn=300010;
char s1[maxn],s2[maxn];
int a[maxn],b[maxn],pow[maxn],rev[maxn],
n,l,t;
int inc(int x,int y)
{
    x+=y;
    return x>=p?x-p:x;
}
int dec(int x,int y)
{
    x-=y;
    return x<0?x+p:x;
}
void ntt(int *a,int flag)
{
    int x,t1,t2;
    for (int i=0;i<l;i++)
        if (rev[i]>i) swap(a[i],a[rev[i]]);
    for (int i=1;i<=t;i++)
        for (int j=0;j<l;j+=1<<i)
        {
            x=0;
            for (int k=j;k<j+(1<<i-1);k++)
            {
                t1=a[k];
                t2=(LL)a[k+(1<<i-1)]*pow[x]%p;
                a[k]=inc(t1,t2);
                a[k+(1<<i-1)]=dec(t1,t2);
                x+=flag*(1<<t-i);
                if (x<0) x+=l;
            }
        }
}
int main()
{
    int inv,u;
    scanf("%d%s%s",&n,s1+1,s2+1);
    for (int i=0;i<n;i++)
    {
        a[i]=s1[n-i]-'0';
        b[i]=s2[n-i]-'0';
    }
    l=1,t=0;
    while (l<n*2-1)
    {
        t++;
        l<<=1;
    }
    pow[0]=pow[1]=1;
    for (int k=(p-1)/l,base=3;k;k>>=1,base=(LL)base*base%p)
        if (k&1) pow[1]=(LL)pow[1]*base%p;
    for (int i=2;i<l;i++) pow[i]=(LL)pow[i-1]*pow[1]%p;
    for (int i=0;i<l;i++)
        for (int j=0;j<t;j++)
            rev[i]|=((i>>j)&1)<<t-j-1;
    ntt(a,1);
    ntt(b,1);
    for (int i=0;i<l;i++) a[i]=(LL)a[i]*b[i]%p;
    ntt(a,-1);
    inv=1;
    for (int k=p-2,base=l;k;k>>=1,base=(LL)base*base%p)
        if (k&1) inv=(LL)inv*base%p;
    for (int i=0;i<l;i++) a[i]=(LL)a[i]*inv%p;
    for (int i=0;i<l;i++)
    {
        a[i+1]+=a[i]/10;
        a[i]%=10;
    }
    if (a[l])
    {
        u=l;
        while (a[u])
        {
            a[u+1]=a[u]/10;
            a[u]%=10;
            u++;
        }
        u--;
    }
    else
    {
        u=l-1;
        while (u&&!a[u]) u--;
    }
    for (int i=u;i>=0;i--) printf("%d",a[i]);
}