Codecraft-18 and Codeforces Round #458 (Div. 1 + Div. 2, combined) G. Sum the Fibonacci（子集卷积）

题目链接：https://codeforces.com/contest/914/problem/G

 

子集卷积+FWT的板子题，记一个子集卷积的板子

 

代码：

#include<bits/stdc++.h>
#define xx first
#define yy second
#define mp make_pair
#define pb push_back
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int MOD=1e9+7;
const int INV2=MOD-MOD/2;
const int MAXN=1e6+5;
const int M=(1<<17)+5;
int s[MAXN],f[MAXN],g[18][M],h[18][M],d[MAXN];
int A[M],B[M],C[M],D[M],cnt[M];
void FMT(int *A,int o)
{
	for(int i=1;i<(1<<17);i<<=1)
		for(int j=0;j<(1<<17);j++)
			if(i&j) (A[j]+=A[j^i]*o)%=MOD;
}
void FWT(int a[],int n,int op)
{
	for(int d=1;d<n;d<<=1)
		for(int m=d<<1,i=0;i<n;i+=m)
			for(int j=0;j<d;j++)
			{
				int x=a[i+j],y=a[i+j+d];
				if(op==0) a[i+j]=(x+y)%MOD,a[i+j+d]=(x-y+MOD)%MOD;
				if(op==1) a[i+j]=(x+y)%MOD;
			}
}
void UFWT(int a[],int n,int op)
{
	for(int d=1;d<n;d<<=1)
		for(int m=d<<1,i=0;i<n;i+=m)
			for(int j=0;j<d;j++)
			{
				int x=a[i+j],y=a[i+j+d];
				if(op==0) a[i+j]=1LL*(x+y)*INV2%MOD,a[i+j+d]=(1LL*(x-y)*INV2%MOD+MOD)%MOD;
				if(op==1) a[i+j]=((x-y)%MOD+MOD)%MOD;
			}
}
int main()
{
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&s[i]);
	f[0]=0,f[1]=1;
	for(int i=2;i<M;i++)
		f[i]=(f[i-1]+f[i-2])%MOD;
	for(int i=1;i<M;i++)
		cnt[i]=__builtin_popcount(i);
	for(int i=1;i<=n;i++)
	{
		d[s[i]]++;g[cnt[s[i]]][s[i]]++;
		B[s[i]]=(B[s[i]]+f[s[i]])%MOD;
	}
	for(int i=0;i<=17;i++) FMT(g[i],1);
	for(int i=0;i<=17;i++)
	{
		for(int j=0;j<=i;j++)
			for(int k=0;k<(1<<17);k++)
				(h[i][k]+=1ll*g[j][k]*g[i-j][k]%MOD)%=MOD;
		FMT(h[i],-1);
		for(int k=0;k<(1<<17);k++) if(cnt[k]!=i) h[i][k]=0;
		//if(i!=17) FMT(h[i],1);
	}
	for(int i=0;i<=17;i++)
		for(int k=0;k<(1<<17);k++)
		{
			 A[k]=(A[k]+1LL*h[i][k]*f[k]%MOD)%MOD;
		}
	FWT(d,(1<<17),0);
	for(int i=0;i<(1<<17);i++)
		d[i]=(1LL*d[i]*d[i])%MOD;
	UFWT(d,(1<<17),0);
	for(int i=0;i<(1<<17);i++)
		C[i]=(C[i]+1LL*d[i]*f[i]%MOD)%MOD;
	FWT(A,(1<<17),1);
	FWT(B,(1<<17),1);
	FWT(C,(1<<17),1);
	for(int i=0;i<(1<<17);i++)
		D[i]=1LL*A[i]*B[i]%MOD*C[i]%MOD;
	UFWT(D,(1<<17),1);
	ll ans=0;
	for(int i=0;i<17;i++)
		ans=(ans+D[1<<i])%MOD;
	printf("%lld\n",ans);
	return 0;
}