C++ 数论相关题目 博弈论:拆分-Nim游戏

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

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

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

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

输出格式
如果先手方必胜,则输出 Yes。

否则,输出 No。

数据范围
1≤n,ai≤100
输入样例:
2
2 3
输出样例:
Yes
C++ 数论相关题目 博弈论:拆分-Nim游戏_第1张图片

#include 
#include 
#include 
#include 

using namespace std;

const int N = 110;
int n;
int f[N];//存i个状态的sg值

int sg(int x)
{
    if(f[x] != -1) return f[x];
    
    unordered_set<int> S; //哈希表存储每个局面可以到的局面
                          //这个地方特别关键:在集合的Nim游戏中,我们可以明显的知道可以到的下一个状态是什么
                          //比如(x - s[i]),这道题里面需要遍历一下所有可能到达的状态,并且异或起来
    for(int i = 0; i < x; i ++ )
        for(int j = 0; j <= i; j ++) //用i和j表示分成的两个状态
            S.insert(sg(i) ^ sg(j));
    
    for(int i = 0; ; i ++ )
        if(!S.count(i))
            return f[x] = i;
}


int main ()
{
    cin>>n;
    
    memset(f, -1, sizeof f); // 记忆化搜索,因为sg值都是自然数,所以初始化成-1,代表没有求过
    
    int res = 0;
    while(n -- )
    {
        int x;
        cin>>x;
        res ^= sg(x);
    }
    
    if(res) puts("Yes");
    else puts("No");
    
    return 0;
}

你可能感兴趣的:(力扣,数论,博弈论,c++,算法)