1500字范文,内容丰富有趣,写作好帮手!
1500字范文 > Codecraft-18 and Codeforces Round #458 (Div. 1 + Div. 2 combined) G. Sum the Fibonacci(子集卷积)

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

时间:2021-12-28 17:43:02

相关推荐

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

题目链接:/contest/914/problem/G

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

代码:

#include<bits/stdc++.h>#define xx first#define yy second#define mp make_pair#define pb push_backusing 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;}

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。