[hdu6057]Kanade‘s convolution

Kanade’s convolution

题解

多简单的一道题呀

我们可以考虑子集求和。毕竟这种又是位运算又是相乘的很容易想到FWT。
由于$(xxory)+(xandy)=(xory)$，所以我们我们可以将枚举的$i,j$改为$x,y$使得$x=iorj,y=ixorj$。
但是由于我们枚举的$x,y$的出现次数是不一样的，所以还要针对的$y$加上一个$2^{bit(y)}$的系数。
原式变成了$C_{x-y}={\sum }_{xandy=y}{2}^{bit(y)}{A}_y{B}_x$。
上式明显可以再把$[xandy=y]$这个条件去掉，根据二进制位来求解，用$k$来代替$x-y$，因为这一位是需要我们枚举的，可化为$C_k={\sum }_{bit(x)-bit(y)=bit(k)}{2}^{bit(y)}{A}_y{B}_x$。这样就可以卷积了。

我们可以先将$2^{bit(y)}$乘入$A_y$中，因为它只与$y$的大小有关。
我们可以采用子集卷积的方法，将$A,B,C$三个多项式根据它们二进制下含1的个数分类，用$F(A,i)$表示$A$中二进制的含1个数为$i$的部分。
有，$F(A,i)={\sum }_{j=0}^{i}F(B,j)F(A,i-j)$。
这明显是可以根据FWT处理的，FWT后相乘再逆回来即可。

时间复杂度为$O\left(nlo{g}^{2}n\right)$。
PS:应该没有人跟我最开始想的一样，先通过二元方程进行转化后跑多项式乘法，最后再解回来的吧。

源码

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define MAXN (1<<19)+5
#define reg register
typedef long long LL;
const int mo=998244353;
const int inv2=499122177;
template<typename _T>
inline void read(_T &x){
	_T f=1;x=0;char s=getchar();
	while('0'>s||'9'<s){if(s=='-')f=-1;s=getchar();}
	while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
	x*=f;
}
int add(const int x,const int y){return x+y<mo?x+y:x+y-mo;}
void FWT(int *A,const int lim,const int typ){
	for(reg int k=1;k<lim;k<<=1)
		for(reg int i=0;i<lim;i+=(k<<1))
			for(reg int j=i;j<i+k;++j){
				A[j]=add(A[j],A[j+k]);
				A[j+k]=add(A[j],add(mo-A[j+k],mo-A[j+k]));
				A[j]=typ^1?1ll*A[j]*inv2%mo:A[j];
				A[j+k]=typ^1?1ll*A[j+k]*inv2%mo:A[j+k];
			}
}
int a[MAXN],b[MAXN],m,lim,bit[MAXN],F[20][MAXN],G[20][MAXN],pw[MAXN],ans[20][MAXN],sum;
signed main(){
	read(m);lim=(1<<m);pw[0]=1;
	for(int i=1;i<=m;i++)pw[i]=add(pw[i-1],pw[i-1]);
	for(int i=1;i<lim;i++)bit[i]=bit[i^(i&-i)]+1;
	for(int i=0;i<lim;i++)read(a[i]),F[bit[i]][i]=1ll*pw[bit[i]]*a[i]%mo;
	for(int i=0;i<lim;i++)read(b[i]),G[bit[i]][i]=b[i];
	for(int i=0;i<=m;i++)FWT(F[i],lim,1),FWT(G[i],lim,1);
	for(int i=0;i<=m;i++)
		for(int j=0;j<=m-i;j++)
			for(int k=0;k<lim;k++)
				ans[i][k]=(1ll*ans[i][k]+1ll*F[j][k]*G[i+j][k])%mo;
	for(int i=0;i<=m;i++)FWT(ans[i],lim,-1);int now=1;
	for(int i=0;i<lim;i++)sum=(1ll*sum+1ll*ans[bit[i]][i]*now)%mo,now=1526ll*now%mo;
	printf("%d\n",sum);
	return 0;
}

谢谢