acwing 博弈论 拆分-Nim游戏

#include<bits/stdc++.h>
using namespace std;
const int N=110;
int n;
int f[N]; //f数组表示每个sg值
//求该题sg函数 
//有多个独立局面，只要单独求每个局面sg值，最后异或起来就行
//主要是求sg函数，这道题，不像集合nim游戏，分成一堆，该题分成两堆。
//sg定理， sg(b1,b2)=sg(b1)^sg(b2);这是一个性质，记住就好。
//求完所有局面的sg的值后，再用mex函数求出这个单独局面的sg值。
//拆分 Nim游戏
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++) //让j<=i,免的重复计算
        {
            S.insert(sg(i)^sg(j));
        }
    }
    for(int i=0;;i++) //以此求mex计算
    {
        if(!S.count(i))
        {
            return f[x]=i;
        }
    }
}
int main()
{
    cin>>n;
    memset(f,-1,sizeof f) ;  //初始化f数组为-1，方便写sg函数判断
    int res=0;
    while(n--)
    {
        int x;
        cin>>x;
        res^=sg(x);
        
    }
    if(res)  cout<<"Yes"<<endl;
    else cout<<"No"<<endl;
    return 0;
}