BZOJ_P1874 [BeiJing2009 WinterCamp]取石子游戏(博弈+SG函数)

BZOJ传送门
Time Limit: 5 Sec Memory Limit: 162 MB
Submit: 670 Solved: 272
[Submit][Status][Discuss]
Description
小H和小Z正在玩一个取石子游戏。 取石子游戏的规则是这样的，每个人每次可以从一堆石子中取出若干个石子，每次取石子的个数有限制，谁不能取石子时就会输掉游戏。 小H先进行操作，他想问你他是否有必胜策略，如果有，第一步如何取石子。

Input
输入文件的第一行为石子的堆数N 接下来N行，每行一个数Ai，表示每堆石子的个数 接下来一行为每次取石子个数的种类数M 接下来M行，每行一个数Bi，表示每次可以取的石子个数，输入保证这M个数按照递增顺序排列。

Output
输出文件第一行为“YES”或者“NO”，表示小H是否有必胜策略。 若结果为“YES”,则第二行包含两个数，第一个数表示从哪堆石子取，第二个数表示取多少个石子，若有多种答案，取第一个数最小的答案，若仍有多种答案，取第二个数最小的答案。

Sample Input
4
7 6 9 3
2
1 2

Sample Output
YES
1 1

Hint
样例中共有四堆石子，石子个数分别为7、6、9、3，每人每次可以从任何一堆石子中取出1个或者2个石子，小H有必胜策略，事实上只要从第一堆石子中取一个石子即可。

数据规模和约定
数据编号 N范围 Ai范围 数据编号 N范围 Ai范围
1 N=2 Ai≤10 6 N≤10 Ai≤10
2 N=2 Ai≤1000 7 N≤10 Ai≤100
3 N=3 Ai≤100 8 N≤10 Ai≤1000
4 N≤10 Ai≤4 9 N≤10 Ai≤1000
5 N≤10 Ai≤7 10 N≤10 Ai≤1000
对于全部数据，M≤10，Bi≤10

HINT

Source
Day2

SG裸题，最后找数的时候枚举就可以了

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
#define N 15
#define M (1<<10)
inline int in(int x=0,char ch=getchar(),int v=1){
    while(ch!='-'&&(ch>'9'||ch<'0')) ch=getchar();if(ch=='-') v=-1,ch=getchar();
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();return x*v;}
int a[N],s[N],mex[M],h[M],n,m,ans,sum;
void GetSG(int n){
    mex[0]=0;
    for(int i=1;i<=n;i++){
        for(int j=1;s[j]<=i&&j<=m;j++)  h[mex[i-s[j]]]=i;
        for(int j=0;j<=n;j++) if(h[j]!=i){mex[i]=j;break;}
    }
}
int main(){
    memset(mex,-1,sizeof(mex));
    n=in();for(int i=1;i<=n;i++) a[i]=in();
    m=in();for(int i=1;i<=m;i++) s[i]=in();
    GetSG(M-1);
    for(int i=1;i<=n;i++) ans^=mex[a[i]];
    if(!ans){puts("NO");return 0;}
    for(int i=1;i<=n;i++){
        sum=ans^mex[a[i]];
        for(int j=1;j<=m&&s[j]<=a[i];j++)
            if(!(sum^mex[a[i]-s[j]])){
                puts("YES");printf("%d %d\n",i,s[j]);
                return 0;
            }
    }
    return 0;
}