Codeforces Round #399 (Div. 1 + Div. 2, combined) - Game of Stones (768E)

题意：给你 n 堆石子，每堆有 ai 个(ai<=60)，每次可选择一堆取若干个，但是对于每一堆石子，去过的数字不能再取，问后手胜败

 

分析：限制条件后，每一堆的选取仍是独立的，而且ai<=60，所以我们可以用状态压缩预先求出SG函数(SG[1]~SG[60])；

 

代码：

#include<map>
#include<cstdio>
#include<iostream>
using namespace std;

const int N = 1E6+10;
typedef long long ll;

map<ll,int>dp[66];
int SG[66];
int n;

int dfs(int x,ll sum){
    //dp[x][sum]表示当前还剩下石子x个，且已经选取的一些数量的二进制集合sum，此状态下的SG函数；
	if(dp[x].count(sum)) return dp[x][sum];  
	ll res=0;
	for(int i=0;i<x;i++){
		if(((sum>>i)&1)==0){  //这个位置没选且可选 
			res|=1<<(dfs(x-i-1,sum|(1ll<<i))); 
		}
	}
	for(int i=0;i<60;i++){   
		if(!(res>>i&1)){   
			dp[x][sum]=i;
			break;
		}
	}
	return dp[x][sum];
}

int main()
{
	for(int i=1;i<=60;i++) SG[i]=dfs(i,0);
    scanf("%d",&n);
	int res=0;
	for(int i=1,x;i<=n;i++){
		scanf("%d",&x);
		res^=SG[x];
	}	
	if(res) puts("NO");
	else puts("YES");
}