CF850E Random Elections

一、题目

点此看题

二、解法

三个候选人其实是一样的，所以我们求出$A$的答案，然后把答案乘$3$就行了。

考虑如果$A-B$，$A-C$这两个投票情况的$i$位（第$i$的选民投的票），如果是$(0,0)$（$A$都不赢），那么就有$BCA$和$CBA$，同理可得，如果是$(1,0)$和$(0,1)$那么$1$种方案，$(1,1)$也是两种方案。综上，这一位异或为$0$就两种情况，否则只有一种。

我们把$f$数组做$xor$卷积，这么做的理由如下：如果两者都是$1$才产生$1$的贡献，这里是乘积（都胜利）；最后我们的下标可以知道选民有多少种方案（算下标化成$n$长度二进制$0$的个数，方案就是$2^{zero}$）

时间复杂度$O(n\log n)$，来个代码。

#include <cstdio>
const int M = 1100005;
const int MOD = 1e9+7;
int read()
{
    int num=0,flag=1;char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n,m,ans,f[M],bit[M],inv2=(MOD+1)/2;
void fwt(int *a,int n,int op)
{
	for(int i=1;i<n;i<<=1)
		for(int p=i<<1,j=0;j<n;j+=p)
			for(int k=0;k<i;k++)
			{
				int x=a[j+k],y=a[i+j+k];
				a[j+k]=(x+y)%MOD;
				a[i+j+k]=(x+MOD-y)%MOD;
				if(op==-1)
				{
					a[j+k]=1ll*a[j+k]*inv2%MOD;
					a[i+j+k]=1ll*a[i+j+k]*inv2%MOD;
				}
			}
}
signed main()
{
	n=read();m=1<<n;
	for(int i=0;i<m;i++)
		scanf("%1d",&f[i]);
	for(int i=1;i<m;i++)
		bit[i]=bit[i>>1]+(i&1);
	fwt(f,m,1);
	for(int i=0;i<m;i++)
		f[i]=1ll*f[i]*f[i]%MOD;
	fwt(f,m,-1);
	for(int i=0;i<m;i++)
		ans=(ans+1ll*(1<<(n-bit[i]))*f[i])%MOD;
	printf("%d\n",ans*3ll%MOD);
}