[CodeForces914G]Sum the Fibonacci

题意:

给你一个长度为$n$ 的数组$s$ .定义五元组$(a,b,c,d,e)$ 是合法的当且仅当:

①.$1\le a,b,c,d,e\le n$

②.$(s_a|s_b)$&$s_c$&$(s_d$ ^$s_e)=2^i,i\in N$

③.$s_a$&$s_b=0$

对于所有合法的五元组$(a,b,c,d,e)$

求$\sum f(s_a|s_b)*f(s_c)*f(s_d$ ^$s_e)\mod 10^9+7$

$f_0=0,f_1=1,f_i=f_{i-1}+f_{i-2}$

$1\le n\le10^6,0\le s_i\lt2^{17}$

---

考虑到$s_i$的范围很小,所以我们考虑记下$cnt[s_i]$表示$s_i$出现的个数

因为题目要求的是通过$(s_a|s_b)$&$s_c$&$(s_d$ ^$s_e)$得到的数

那么$\sum f(s_a|s_b)*f(s_c)*f(s_d$ ^$s_e)$

$=cnt[s_a|s_b]*f(s_a|s_b)$&$cnt[s_c]*f(s_c)$&$cnt[s_d$^$s_e]*f(s_d$^$s_e)$

这里的&表示与卷积

考虑$cnt[s_a|s_b]=\sum[s_a$&$s_b=0]cnt[s_a]*cnt[s_b]$

其实这就是子集卷积的另外一种表达方式

不会的看$vfk2015$的集训队论文或者右转这里

$cnt[s_d$^$s_e]$这个就是$cnt[s]$与自己的异或卷积

最后把所有的$cnt$都求出来之后再乘上相应位置的$f$

然后求一遍与卷积,把$2^i$次方项的答案加起来就好了

以上的与卷积,异或卷积(还有或卷积)都可以用$FWT$(快速沃尔什变换)求出

这个$FWT$的话我们可以很好的通过背代码理解他

#include<bits/stdc++.h>
#define fp(i,a,b) for(register int i=a,I=b+1;i<I;++i)
#define fd(i,a,b) for(register int i=a,I=b-1;i>I;--i)
#define go(i,u) for(register int i=fi[u],v=e[i].to;i;v=e[i=e[i].nx].to)
#define file(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)
template<class T>inline bool cmax(T&a,const T&b){return a<b?a=b,1:0;}
template<class T>inline bool cmin(T&a,const T&b){return a>b?a=b,1:0;}
using namespace std;
char ss[1<<17],*A=ss,*B=ss;
inline char gc(){return A==B&&(B=(A=ss)+fread(ss,1,1<<17,stdin),A==B)?-1:*A++;}
template<class T>inline void sd(T&x){
    char c;T y=1;while(c=gc(),(c<48||57<c)&&c!=-1)if(c==45)y=-1;x=c-48;
    while(c=gc(),47<c&&c<58)x=x*10+c-48;x*=y;
}
char sr[1<<21],z[20];int C=-1,Z;
inline void Ot(){fwrite(sr,1,C+1,stdout),C=-1;}
template<class T>inline void we(T x){
    if(C>1<<20)Ot();if(x<0)sr[++C]=45,x=-x;
    while(z[++Z]=x%10+48,x/=10);
    while(sr[++C]=z[Z],--Z);sr[++C]='\n';
}
const int N=19,M=1<<17|1,P=1e9+7,inv=5e8+4;
typedef int arr[M];
int n,S,ans,Mi[N];arr a[N],b[N],c,d,e,f,p[N],t,pos,cnt;
inline int add(int a,int b){return a+=b,a>=P?a-P:a;}
inline int sub(int a,int b){return a-=b,a<0?a+P:a;}
inline void fmt(int*a,int op(int,int)){
    fp(i,1,n)fp(s,1,S)if(s&Mi[i])
        a[s]=op(a[s],a[s^Mi[i]]);
}
inline void fwt(int*a,int op){
    fp(i,1,n)fp(s,1,S)
        if(s&Mi[i]){
        int o=a[s],t=Mi[i];
        if(op==1)a[s-t]=add(a[s-t],o);
        else if(op==2)a[s]=sub(a[s-t],o),a[s-t]=add(a[s-t],o);
        else a[s]=add(a[s-t],o);
    }
}
inline void ifwt(int*a,int op){
    fp(i,1,n)fp(s,1,S)if(s&Mi[i]){
        int o=a[s],t=Mi[i];
        if(op==1)a[s-t]=sub(a[s-t],o);
        else if(op==2)a[s]=1ll*sub(a[s-t],o)*inv%P,a[s-t]=1ll*add(a[s-t],o)*inv%P;
        else a[s]=sub(o,a[s-t]);
    }
}
int main(){
    #ifndef ONLINE_JUDGE
        file("s");
    #endif
    pos[1]=Mi[1]=1;fp(i,2,18)pos[Mi[i]=Mi[i-1]<<1]=i;
    sd(n);int x,y=0;fp(i,1,n)sd(x),++t[x],cmax(y,x);
    S=1;while(S<=y)S<<=1;n=pos[S--];
    fp(i,1,S)cnt[i]=cnt[i>>1]+(i&1);
    f[1]=1;fp(i,2,S)f[i]=add(f[i-1],f[i-2]);
    fp(s,0,S)a[cnt[s]][s]=b[cnt[s]][s]=d[s]=t[s],c[s]=1ll*t[s]*f[s]%P;;
    fwt(d,2);fp(s,0,S)d[s]=1ll*d[s]*d[s]%P;
    ifwt(d,2);fp(s,0,S)d[s]=1ll*d[s]*f[s]%P;
    fp(i,0,n)fmt(a[i],add),fmt(b[i],add);
    fp(i,0,n)fp(j,0,n-i)fp(s,0,S)p[i+j][s]=add(p[i+j][s],1ll*a[i][s]*b[j][s]%P);
    fp(i,0,n)fmt(p[i],sub);fp(s,0,S)e[s]=1ll*p[cnt[s]][s]*f[s]%P;
    fwt(c,1),fwt(d,1),fwt(e,1);
    fp(s,0,S)c[s]=1ll*c[s]*d[s]%P*e[s]%P;
    ifwt(c,1);fp(i,1,n)ans=add(ans,c[Mi[i]]);
    printf("%d",ans);
return Ot(),0;
}