HDU6057 Kanade‘s convolution

最新推荐文章于 2024-10-05 12:07:50 发布

tanjunming2020

最新推荐文章于 2024-10-05 12:07:50 发布

阅读量1.4k

点赞数 3

分类专栏：题解文章标签： c++ 题解

本文链接：https://blog.csdn.net/tanjunming2020/article/details/130414152

版权

题解专栏收录该内容

348 篇文章 0 订阅

订阅专栏

HDU6057 Kanade’s convolution

题目大意

给你两个长度为 $2^n-1$ 的数组 $A$ 和 $B$ ，计算长度为 $2^n-1$ 的数组 $C$

$C[k]=\sum\limits_{i \ and \ j =k}A[i \ xor \ j]\times B[i \ or \ j]$

你需要输出 $\sum\limits_{i=0}^{2^m-1}C[i]\times 1526^i$ 对 $998244353$ 取模后的值。

$1\leq m\leq 19,0\leq A[i],B[i]<998244353$

题解

前置知识：快速沃尔什变换（ $F W T$ ）

令 $x=i\oplus j,y=i|j$ ，因为 $(i|j)-(i\& j)=i\oplus j$ ，所以 $i\&j=(i|j)-(i\oplus j)$ ，则

$C_k=\sum\limits_{y-x=k,x\& k=0}2^{d_x}\times A_x\times B_y$

其中 $d_i$ 表示 $i$ 的二进制位中为 $1$ 的位数。

因为对于一对数对 $(x, y)$ ， $x$ 的每一个为 $1$ 的位， $i, j$ 都有两种选法（ $i$ 这一位为 $1$ 且 $j$ 这一位为 $0$ 、 $i$ 这一位为 $0$ 且 $j$ 这一位为 $1$ ）使 $x=i\oplus j,y=i|j$ 。所以有 $2^{d_x}$ 对 $(i, j)$ 满足上述条件，要乘上 $2^{d_x}$ 。

因为 $x=i\oplus j,y=i|j$ ，所以 $x\&y=y$ ，可将 $y - x = k$ 换成 $y\oplus x=k$ 。 $x\& k=0$ 也可表示为 $d_x+d_k=d_y$ 。再令 $A_i=A_i\times 2^{d_i}$ ，则

$C_k=\sum\limits_{y\oplus x=k,d_x+d_k=d_y}A_x\times B_y$

多项式 $V\!A_i$ 表示二进制位为 $1$ 的个数为 $i$ 的 $A$ 值，多项式 $V\!B_i$ 表示二进制位为 $1$ 的个数为 $i$ 的 $B$ 值，二进制位为 $1$ 的个数不等于 $i$ 的数在 $V\!A_i,V\!B_i$ 中值为 $0$ 。

对于每一个 $i$ ，对 $V\!A_i$ 和 $V\!B_i$ 做一次异或卷积的 $F W T$ 。令 $V_k=\sum\limits_{j=0}^i(V\!A_j)_k\times (V\!B_{i-j})_k$ ，将 $V_k$ 逆变换回来。若 $k$ 中二进制位为 $1$ 的个数为 $i$ ，则 $C_k=V_k$ 。

最后，对于每个 $C_i$ 都乘上 $1526^i$ 即可。

时间复杂度为 $O(n\log^2 n)$ 。

code

#include<iostream>
#include<cstdio>
using namespace std;
int n,cnt[1<<20];
long long x,ans=0,now=1,ta[22][1<<20],tb[22][1<<20],c[1<<20],v[1<<20];
const int mod=998244353,ny2=499122177;
void fwt(long long *w,int fl){
	for(int s=2;s<=1<<n;s<<=1){
		int mid=s>>1;
		for(int v=0;v<1<<n;v+=s){
			for(int i=0;i<mid;i++){
				int t1=w[v+i],t2=w[v+mid+i];
				w[v+i]=(t1+t2)%mod;w[v+mid+i]=(t1-t2+mod)%mod;
				if(fl==-1){
					w[v+i]=1ll*w[v+i]*ny2%mod;
					w[v+mid+i]=1ll*w[v+mid+i]*ny2%mod;
				}
			}
		}
	}
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<1<<n;i++) cnt[i]=cnt[i-(i&(-i))]+1;
	for(int i=0;i<1<<n;i++){
		scanf("%lld",&x);
		x=x*(1ll<<cnt[i])%mod;
		ta[cnt[i]][i]=x;
	}
	for(int i=0;i<1<<n;i++){
		scanf("%lld",&x);
		tb[cnt[i]][i]=x;
	}
	for(int i=0;i<=n;i++){
		fwt(ta[i],1);fwt(tb[i],1);
	}
	for(int i=0;i<=n;i++){
		for(int j=0;i+j<=n;j++){
			for(int k=0;k<1<<n;k++){
				c[k]=(c[k]+1ll*ta[j][k]*tb[i+j][k]%mod)%mod;
			}
		}
		fwt(c,-1);
		for(int j=0;j<1<<n;j++){
			if(cnt[j]==i) v[j]=c[j];
			c[j]=0;
		}
	}
	for(int i=0;i<1<<n;i++){
		ans=(ans+1ll*v[i]*now%mod)%mod;
		now=now*1526ll%mod;
	}
	printf("%lld\n",ans);
	return 0;
}