bzoj 1874: [BeiJing2009 WinterCamp]取石子游戏 Nim游戏+SG函数

题意

有n堆石子，第i堆石子有a[i]块石头。每次只能在一堆石子中取b[j]个{1<=j<=m}
问先取者是否有必胜策略，有则输出第一次在第几堆石子取多少个。
n,m<=10,a[i]<=1000

分析

第一次接触SG函数，感觉是个蛮神奇的东西。

转载一点相关的文章：
首先定义mex(minimal excludant)运算，这是施加于一个集合的运算，表示最小的不属于这个集合的非负整数。例如mex{0,1,2,4}=3、mex{2,3,5}=0、mex{}=0。
对于一个给定的有向无环图，定义关于图的每个顶点的Sprague-Grundy函数g如下：g(x)=mex{ g(y) | y是x的后继 },这里的g（x）即sg[x]
例如：取石子问题，有1堆n个的石子，每次只能取{1，3,4}个石子，先取完石子者胜利，那么各个数的SG值为多少？
sg[0]=0，f[]={1,3,4},
x=1时，可以取走1-f{1}个石子，剩余{0}个，mex{sg[0]}={0}，故sg[1]=1;
x=2时，可以取走2-f{1}个石子，剩余{1}个，mex{sg[1]}={1}，故sg[2]=0；
x=3时，可以取走3-f{1,3}个石子，剩余{2,0}个，mex{sg[2],sg[0]}={0,0}，故sg[3]=1;
x=4时，可以取走4-f{1,3,4}个石子，剩余{3,1,0}个，mex{sg[3],sg[1],sg[0]}={1,1,0},故sg[4]=2;
x=5时，可以取走5-f{1,3,4}个石子，剩余{4,2,1}个，mex{sg[4],sg[2],sg[1]}={2,0,1},故sg[5]=3；
以此类推…..
x 0 1 2 3 4 5 6 7 8….
sg[x] 0 1 0 1 2 3 2 0 1….

知道SG函数之后这题就变得很简单了。
首先最经典的Nim游戏是若a[1]^a[2]^…^a[n]!=0则先手有必胜策略，推广到这题，若sg[a[1]]^sg[a[2]]^…^sg[a[n]]!=0则先手有必胜策略，证明同普通的Nim游戏类似，在此略过。
最后枚举一下即可。

代码

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>#define N 15using namespace std;int n,m,a[N],f[N],sg[1005],hash[1005];void getsg(int n,int m){    for (int i=1;i<=n;i++)    {        for (int j=1;j<=m;j++)            if (i-f[j]>=0) hash[sg[i-f[j]]]=i;        for (int j=0;j<=i;j++)            if (hash[j]!=i)            {                sg[i]=j;                break;            }    }}int main(){    scanf("%d",&n);    for (int i=1;i<=n;i++)        scanf("%d",&a[i]);    scanf("%d",&m);    for (int i=1;i<=m;i++)        scanf("%d",&f[i]);    getsg(1000,m);    int sum=0;    for (int i=1;i<=n;i++)        sum^=sg[a[i]];    if (!sum)    {        printf("NO");        return 0;    }    printf("YES\n");    for (int i=1;i<=n;i++)    {        int tmp=sg[a[i]]^sum;        for (int j=1;j<=m;j++)            if (a[i]-f[j]>=0)                if ((tmp^sg[a[i]-f[j]])==0)                {                    printf("%d %d",i,f[j]);                    return 0;                }    }    return 0;}