正题
题目链接:https://www.luogu.com.cn/problem/P6097
题目大意
长度为2n2^n2n的序列a,ba,ba,b求一个ccc满足
ck=∑i∣j=k,i&j=∅ai×bjc_k=\sum_{i|j=k,i\&j=\varnothing}a_i\times b_jck=i∣j=k,i&j=∅∑ai×bj
解题思路
从炫酷反演魔术过来的,顺便写掉这题
简单的说就是求kkk的所有子集和其补集的乘积和。
只有i∣j=ki|j=ki∣j=k的话就是普通的FWT\text{FWT}FWT了,但是还有i&j=∅i\&j=\varnothingi&j=∅这个东西。
一个巧妙的想法是把这个条件转换为∣i∣+∣j∣=∣i∪j∣|i|+|j|=|i\cup j|∣i∣+∣j∣=∣i∪j∣,显然两个之间是充要的。
然后可以把aia_iai存在fct(i),if_{ct(i),i}fct(i),i这个位置,其中ct(i)ct(i)ct(i)表示iii中111的个数。同理bib_ibi存在gct(i),ig_{ct(i),i}gct(i),i这个位置。
然后就有卷积
ha,b=∑i+j=a,x∣y=bfi,x×gj,yh_{a,b}=\sum_{i+j=a,x|y=b}f_{i,x}\times g_{j,y}ha,b=i+j=a,x∣y=b∑fi,x×gj,y
这个先暴力FWT\text{FWT}FWT了f,gf,gf,g然后暴力卷积然后IFWT\text{IFWT}IFWT回去就好了。
时间复杂度O(n22n)O(n^22^n)O(n22n),不能全开long long\text{long long}long long不然会T\text TT的
code
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=21,P=1e9+9;
int n,k,ct[1<<N],f[N][1<<N],g[N][1<<N],h[N][1<<N];
void FWT(int *f,int op){for(int p=2;p<=n;p<<=1)for(int k=0,len=p>>1;k<n;k+=p)for(int i=k;i<k+len;i++)(f[i+len]+=(f[i]*op+P)%P)%=P;return;
}
signed main()
{// printf("%d",sizeof(f)>>20);scanf("%d",&k);n=(1<<k);for(int i=0;i<n;i++){if(i)ct[i]=ct[i-(i&-i)]+1;scanf("%d",&f[ct[i]][i]);}for(int i=0;i<n;i++)scanf("%d",&g[ct[i]][i]);for(int i=0;i<=k;i++)FWT(f[i],1),FWT(g[i],1);for(int i=0;i<=k;i++)for(int j=0;j<=i;j++)for(int x=0;x<n;x++)(h[i][x]+=1ll*f[j][x]*g[i-j][x]%P)%=P;for(int i=0;i<=k;i++)FWT(h[i],-1);for(int i=0;i<n;i++)printf("%d ",h[ct[i]][i]);return 0;
}