【BZOJ1874】取石子游戏（SG函数）

题意：小H和小Z正在玩一个取石子游戏。 取石子游戏的规则是这样的，每个人每次可以从一堆石子中取出若干个石子，

每次取石子的个数有限制，谁不能取石子时就会输掉游戏。 小H先进行操作，

他想问你他是否有必胜策略，如果有，第一步如何取石子

n<=10,a[i]<=1000,m<=10,b[i]<=10

思路：求一下SG函数再枚举一下方案……

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
typedef long long ll;
using namespace std;
#define N   21000
#define oo  10000000
#define MOD 1000000007

int a[N],b[N],c[N],flag[N],g[N],n;

int isok(int x,int y)
{
    a[x]-=y;
    int ans=0;    
    for(int i=1;i<=n;i++) ans^=g[a[i]];
    a[x]+=y;
    if(ans) return 0;
    return 1;
}

int main()
{ 
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    int m;
    scanf("%d",&m);
    memset(c,0,sizeof(c));
    for(int i=1;i<=m;i++) 
    {
        scanf("%d",&b[i]);
        c[b[i]]=1;
    }
    g[0]=0;
    for(int i=1;i<=1000;i++)
    {
        memset(flag,0,sizeof(flag));
        for(int j=1;j<=m;j++)
         if(i-b[j]>=0) flag[g[i-b[j]]]=1;
        int j=0;
        while(flag[j]) j++;
        g[i]=j;
    }
    int ans=0;
    for(int i=1;i<=n;i++) ans^=g[a[i]];
    if(ans) 
    {
        int s1=0; int s2=0;
        for(int i=1;i<=n;i++)
         for(int j=1;j<=a[i];j++)
          if(s1+s2==0&&c[j]&&isok(i,j)){s1=i; s2=j;}
        printf("YES
");
        printf("%d %d
",s1,s2);
    }
     else printf("NO
");
    return 0;
}