HDU 6057 Kanade's convolution（FWT）

Description

Input
第一行一整数m，之后输入序列A和B（m<=19,0<=A[i],B[i]<998244353）
Output
输出答案
Sample Input
2
1 2 3 4
5 6 7 8
Sample Output
568535691
Solution

Code

#include<cstdio>#include<cstring>using namespace std;typedef long long ll;typedef unsigned long long ull;const int mod=998244353,inv2=499122177,maxn=1<<20;int m,n,A[maxn],B[maxn],C[maxn],bit[maxn],a[maxn][21],b[maxn][21],c[maxn][21];void read(int &x){    x=0;    char p=getchar();    while(!(p<='9'&&p>='0'))p=getchar();    while(p<='9'&&p>='0')x*=10,x+=p-48,p=getchar();}int mod_pow(int a,int b){    int ans=1;    while(b)    {        if(b&1)ans=(ll)ans*a%mod;        a=(ll)a*a%mod;        b>>=1;    }    return ans;}void FWT(int a[maxn][21],int n,int sta)  {      for(int d=1;d<n;d<<=1)          for(int i=0;i<n;i+=(d<<1))              for(int j=0;j<d;j++)                  for(int k=0;k<=m;k++)                {                      int x=a[i+j][k],y=a[i+j+d][k];                      a[i+j][k]=(x+y)%mod,a[i+j+d][k]=(x-y+mod)%mod;                     //xor:a[i+j]=x+y,a[i+j+d]=(x-y+mod)%mod;                      //and:a[i+j]=x+y;                      //or:a[i+j+d]=x+y;                  }     if(sta==1)    {        int inv=mod_pow(inv2,m);        for(int i=0;i<n;i++)            for(int j=0;j<=m;j++)                a[i][j]=(ll)a[i][j]*inv%mod;    }}    ull temp[21];int main(){    read(m);    n=1<<m;    for(int i=0;i<n;i++)read(A[i]);    for(int i=0;i<n;i++)read(B[i]);    for(int i=0;i<n;i++)bit[i]=bit[i>>1]+(i&1);    for(int i=0;i<n;i++)A[i]=(ll)A[i]*(1<<bit[i])%mod;    for(int i=0;i<n;i++)a[i][bit[i]]=A[i],b[i][bit[i]]=B[i];    FWT(a,n,0),FWT(b,n,0);    for(int i=0;i<n;i++)    {        memset(temp,0,sizeof(temp));        for(int j=0;j<=m;j++)            for(int k=0;k<=j;k++)            {                //c[i][j-k]=(c[i][j-k]+(ll)a[i][k]*b[i][j])%mod;                temp[j-k]+=(ll)a[i][k]*b[i][j];                if(temp[j-k]>=(1ll<<63))temp[j-k]%=mod;            }        for(int j=0;j<=m;j++)c[i][j]=temp[j]%mod;    }    FWT(c,n,1);    for(int i=0;i<n;i++)C[i]=c[i][bit[i]];    int ans=0,p=1;    for(int i=0;i<n;i++)    {        ans=(ans+(ll)C[i]*p)%mod;        p=1526ll*p%mod;    }    printf("%d\n",ans);    return 0;}