取球游戏蓝桥杯python简单博弈sg函数

题目描述

今盒子里有 n 个小球，A、B 两人轮流从盒中取球，每个人都可以看到另一个人取了多少个，也可以看到盒中还剩下多少个，并且两人都很聪明，不会做出错误的判断。

我们约定：

每个人从盒子中取出的球的数目必须是：1，3，7 或者 8 个。轮到某一方取球时不能弃权！A 先取球，然后双方交替取球，直到取完。被迫拿到最后一个球的一方为负方（输方）

请编程确定出在双方都不判断失误的情况下，对于特定的初始球数，A 是否能赢？

输入描述

先是一个整数 n (n<100)，表示接下来有 n 个整数。

然后是 n 个整数，每个占一行（整数<10e4），表示初始球数。

输出描述

程序则输出 n 行，表示 A 的输赢情况（输为 0，赢为 1）。

输入输出样例

示例

输入

４
１
２
10
18

输出

0
1
1
0

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

思路：注意是谁最后取谁输，所以初始化vis[1]=0

import sys
sys.setrecursionlimit(10000000)
vis=[-1 for i in range(10100)]
vis[1]=0
vis[0]=0
def sg(x):
    if (vis[x]!=-1):
        return vis[x]
    step=[1,3,7,8]
    v=[False for i in range(1010)]
    for i in step:
        if (x-i)<=0:
            break
        mid=sg(x-i)
        v[mid]=True
    i=0
    while (True):
        if (v[i]==False):
            break
        i+=1
    vis[x]=i
    return vis[x]
sg(10000)
#print(vis)


n=int(input())
for i in range(n):
    mid=int(input())
    if (vis[mid]!=0):
        print(1)
    else:
        print(0)