题目描述
题解
假设A连赢两场,第一次得到的状态是 s 1 s1 s1 ,第二次得到的状态是 s 2 s2 s2 ,对于第 i i i 个人来说,如果两次的状态分别是 00 00 00 或 11 11 11 的话,那就有两种方案,如果是 01 01 01 或 10 10 10 的话那就只有一种方案,所以我们可以把 s 1 ∧ s 2 s1 \wedge s2 s1∧s2 ,它对答案的贡献就是 2 c n t 0 2^{cnt_0} 2cnt0 ,于是我们把 a a a 数组自卷积,对于状态 s s s 乘上上述值即可,答案记得乘 3 3 3
代码
#include <bits/stdc++.h>
using namespace std;
const int N=1<<20,P=1e9+7,I=(P+1)>>1;
int n,t,a[N],b[N],A;char s[N];
int X(int x){return x>=P?x-P:x;}
void fwt(int *g,int o){
for (int i=1;i<t;i<<=1)
for (int j=0;j<t;j+=(i<<1))
for (int x,y,k=0;k<i;k++){
x=g[j+k];y=g[i+j+k];
g[j+k]=X(x+y);g[i+j+k]=X(x-y+P);
if (o) g[j+k]=1ll*g[j+k]*I%P,
g[i+j+k]=1ll*g[i+j+k]*I%P;
}
}
int main(){
scanf("%d%s",&n,s);t=1<<n;
for (int i=0;i<t;i++)
a[i]=(s[i]^48),
b[i]=b[i>>1]+(i&1);fwt(a,0);
for (int i=0;i<t;i++)
a[i]=1ll*a[i]*a[i]%P;fwt(a,1);
for (int i=0;i<t;i++)
A=X(A+1ll*a[i]*(1<<(n-b[i]))%P);
cout<<3ll*A%P<<endl;return 0;
}