AcWing 894. 拆分-Nim游戏 (博弈+记忆化搜索)

题目链接：点此跳转

给定n堆石子，两位玩家轮流操作，每次操作可以取走其中的一堆石子，然后放入两堆规模更小的石子（新堆规模可以为0，且两个新堆的石子总数可以大于取走的那堆石子数），最后无法进行操作的人视为失败。

问如果两人都采用最优策略，先手是否必胜。

输入格式
第一行包含整数n。

第二行包含n个整数，其中第i个整数表示第i堆石子的数量ai。

输出格式
如果先手方必胜，则输出“Yes”。

否则，输出“No”。

解题思路：

运用SG函数递归求解，枚举每次操作后形成的石子局面 插入到mex集合中，最后枚举i求解mex集合即可

Code：

#include<iostream>
#include<algorithm>
#include<unordered_set>
#include<cstring>

using namespace std;
const int maxn=1007;

int f[maxn];

int sg(int x){
    if(f[x]!=-1) return f[x];
    
    unordered_set<int> S;
    for(int i=0;i<x;i++){
        for(int j=0;j<=i;j++){
            S.insert(sg(i)^sg(j));
        }
    }
    
    //mex集合操作
    for(int i=0;;i++){
        if(!S.count(i)) return f[x]=i;
    }
}

int main(){
    
    memset(f,-1,sizeof f);
    int res=0;
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        
        int x;
        scanf("%d",&x);
        res^=sg(x);
    }
    
    if(res) puts("Yes");
    else puts("No");
}