题解 | #牛牛的石头游戏#

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param piles int整型一维数组 
# @return int整型一维数组
#
class Solution:
    def canWin(self , piles: List[int]) -> List[int]:
        # write code here
        n=max(piles)
        dp = [0,1,1,1,0]
        res=[]
        for i in range(5,n+1):
            dp.append(1-min(dp[i-3:i]))

        for pile in piles:
            res.append(dp[pile])
        return res