【博弈】【清华冬令营2018模拟】取石子

题目

Description
有n堆石子，第i堆有xi个。
Alice和Bob轮流取石子（先后手未定），Alice每次从一堆中取走a个，Bob每次从一堆中取走b个，无法操作者输。
不难发现只会有四种情况：Alice必胜；Bob必胜；先手必胜；后手必胜。
你需要选定若干堆石子（共有2^n种方案），Alice和Bob只能在你选出的堆中取，问以上四种情况对应的方案数。

Input
第一行三个整数n,a,b，第二行n个整数x1~xn。

Output
一行四个整数，分别表示Alice必胜、Bob必胜、先手必胜和后手必胜的方案数，对10^9+7取模。

Sample Input
2 2 3
2 3

Sample Output
2 0 1 1

【样例解释】
选定空集时后手必胜，选定{2}时Alice必胜，选定{3}时先手必胜，选定{2,3}时Alice必胜。

Data Constraint
对于10%的数据，n,xi<=5。
对于50%的数据，n<=20。
对于另外10%的数据，a=b。
对于又另外20%的数据，a=1。
对于100%的数据，1<=n<=100000，1<=a,b,xi<=10^9。

思路

先令a<=b
把石子堆按照%(a+b)的余数r分类
一： 0<=r<=a-1
二： a<=r<=b-1
三： b<=r<=2a-1
四： 2a<=r<=a+b-1
显然对于a+b双方会各取一个
然后发现一对胜负没有意义
有二则Alice必胜
每一个三会改变一次先后顺序
当有两个及以上四的时候Alice同样必胜
而只有一个四 则Alice先手必胜 Bob先手相当于一个三
同样有显然结论为Bob不可能必胜
那么看起来相对难算的Alice必胜可以由pow(2,n)-先后手必胜得到
令F为有奇数个三的方案数 G为有偶数个三的方案数
其实就是 F=C(1,三)+C(3,三)+… G=C(0,三)+C(2,三)+…
后手必胜就是 pow(2,一)G
先手必胜就是 pow(2,一)(F+G*四)
后手必胜中没有四的原因是有四的情况Alice先手必胜

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1e5+77,mod=1e9+7;
ll a[N],n,A,B,s1,s2,s3,bz=0,ans[4],e[N];
int main()
{
	freopen("stone.in","r",stdin); freopen("stone.out","w",stdout);
	scanf("%lld%lld%lld",&n,&A,&B);
	e[0]=1;
	for(int i=1; i<=n; i++) e[i]=(e[i-1]*2)%mod;
	for(int i=1; i<=n; i++) scanf("%lld",&a[i]),a[i]%=(A+B);
	if(A>B) swap(A,B),bz=1;
	for(int i=1; i<=n; i++)
	{
		if(a[i]<A) s1++;
		if(a[i]>=B&&a[i]<A*2) s2++;
		if(a[i]>=max(B,2*A)) s3++;
	}
	if(s2==0) ans[2]=s3*e[s1],ans[3]=e[s1];
	else ans[2]=e[s2-1]*(s3+1)%mod*e[s1],ans[3]=e[s2-1]*e[s1];
	ans[2]%=mod;ans[3]%=mod;
	ans[bz]=(e[n]-ans[2]-ans[3]+mod)%mod;
	printf("%lld %lld %lld %lld\n",ans[0],ans[1],ans[2],ans[3]);
}