【CF914G】Sum the Fibonacci
题解:给你一个长度为n的数组s。定义五元组(a,b,c,d,e)是合法的当且仅当:
1. $1\le a,b,c,d,e\le n$
2. $(s_a|s_b) \& s_c \& (s_d $^$ s_e)=2^i$,i是某个整数
3. $s_a \& s_b=0$
求$\sum f(s_a|s_b) * f(s_c) * f(s_d $^$ s_e)$,f是斐波那契数列,对于所有合法的五元组(a,b,c,d,e)。答案模$10^9+7$。
$1\le n\le 10^6,0\le s_i< 2^17$
题解:说白了就是求:子集和卷积,异或卷积,与卷积。后面两个好求,学了一发子集和卷积。说白了就是强行加了一个占位多项式,即将数组多开一维记录子集中1的个数。然后合并时相当于背包合并,时间复杂度$O(n^22^n)$。
#include <cstdio> #include <cstring> #include <iostream> using namespace std; typedef long long ll; const int maxn=(1<<17)+4; const ll P=1000000007; const ll inv=500000004; ll a[maxn],c[maxn],d[maxn]; ll fa[maxn][18],fb[maxn][18],f[maxn]; ll ans; int cnt[maxn]; int n,len; inline int rd() { int ret=0,f=1; char gc=getchar(); while(gc<‘0‘||gc>‘9‘) {if(gc==‘-‘) f=-f; gc=getchar();} while(gc>=‘0‘&&gc<=‘9‘) ret=ret*10+gc-‘0‘,gc=getchar(); return ret*f; } int main() { n=rd(); int h,i,j,k,x,y,v; f[1]=cnt[1]=1; len=1<<17; for(i=2;i<len;i++) f[i]=(f[i-1]+f[i-2])%P,cnt[i]=cnt[i-(i&-i)]+1; for(i=1;i<=n;i++) v=rd(),fa[v][cnt[v]]++,d[v]++,c[v]=(c[v]+f[v])%P; for(h=1;h<len;h<<=1) for(i=0;i<len;i++) if(i&h) for(j=0;j<=17;j++) fa[i][j]=(fa[i][j]+fa[i-h][j])%P; for(i=0;i<len;i++) for(j=0;j<=17;j++) for(k=0;k<=j;k++) fb[i][j]=(fb[i][j]+fa[i][k]*fa[i][j-k])%P; for(h=1;h<len;h<<=1) for(i=0;i<len;i++) if(i&h) for(j=0;j<=17;j++) fb[i][j]=(fb[i][j]-fb[i-h][j]+P)%P; for(i=0;i<len;i++) a[i]=(a[i]+f[i]*fb[i][cnt[i]])%P; for(h=1;h<len;h<<=1) for(i=0;i<len;i++) if(i&h) x=d[i-h],y=d[i],d[i-h]=(x+y)%P,d[i]=(x-y+P)%P; for(i=0;i<len;i++) d[i]=d[i]*d[i]%P; for(h=1;h<len;h<<=1) for(i=0;i<len;i++) if(i&h) x=d[i-h],y=d[i],d[i-h]=(x+y)*inv%P,d[i]=(x-y+P)*inv%P; for(i=0;i<len;i++) d[i]=d[i]*f[i]%P; for(h=1;h<len;h<<=1) for(i=0;i<len;i++) if(!(i&h)) a[i]=(a[i]+a[i+h])%P,c[i]=(c[i]+c[i+h])%P,d[i]=(d[i]+d[i+h])%P; for(i=0;i<len;i++) a[i]=a[i]*c[i]%P*d[i]%P; for(h=1;h<len;h<<=1) for(i=0;i<len;i++) if(!(i&h)) a[i]=(a[i]-a[i+h]+P)%P; for(i=0;i<17;i++) ans=(ans+a[1<<i])%P; printf("%lld",ans); return 0; }