[BZOJ1874][BeiJing2009 WinterCamp]取石子游戏（博弈SG函数）

题目描述

传送门

题解

求SG函数就不说了
不过我犯了一个非常智障的错误…
一般的Nim游戏判断必胜策略的方法应该是看sg(i)>k^sg(i)，但是这里只是判断是否能从这一堆拿，而不是拿几个
判断拿几个的话就必须让xor值为0去到达必败态

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define N 10005

int n,m,k;
int a[N],rule[N],sg[N];

int get_sg(int x)
{
    if (sg[x]!=-1) return sg[x];
    bool ext[N];memset(ext,0,sizeof(ext));
    for (int i=1;i<=m;++i)
    {
        if (x<rule[i]) break;
        ext[get_sg(x-rule[i])]=1;
    }
    for (int i=0;;++i)
        if (!ext[i]) {sg[x]=i;break;}
    return sg[x];
}
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",&rule[i]);
    sort(rule+1,rule+m+1);
    memset(sg,-1,sizeof(sg));
    for (int i=1;i<=n;++i)
        k^=get_sg(a[i]);
    if (!k) {puts("NO");return 0;}
    for (int i=1;i<=n;++i)
    {
        for (int j=1;j<=m;++j)
        {
            if (a[i]<rule[j]) break;
            if ((k^get_sg(a[i])^get_sg(a[i]-rule[j]))==0)
            {
                puts("YES");
                printf("%d %d\n",i,rule[j]);
                return 0;
            }
        }
    }
}