算法60---石子游戏【动态规划】

一、题目：

亚历克斯和李用几堆石子在做游戏。偶数堆石子排成一行，每堆都有正整数颗石子 piles[i] 。

游戏以谁手中的石子最多来决出胜负。石子的总数是奇数，所以没有平局。

亚历克斯和李轮流进行，亚历克斯先开始。 每回合，玩家从行的开始或结束处取走整堆石头。 这种情况一直持续到没有更多的石子堆为止，此时手中石子最多的玩家获胜。

假设亚历克斯和李都发挥出最佳水平，当亚历克斯赢得比赛时返回 true ，当李赢得比赛时返回 false 。

 

示例：

输入：[5,3,4,5]
输出：true
解释：
亚历克斯先开始，只能拿前 5 颗或后 5 颗石子 。
假设他取了前 5 颗，这一行就变成了 [3,4,5] 。
如果李拿走前 3 颗，那么剩下的是 [4,5]，亚历克斯拿走后 5 颗赢得 10 分。
如果李拿走后 5 颗，那么剩下的是 [3,4]，亚历克斯拿走后 4 颗赢得 9 分。
这表明，取前 5 颗石子对亚历克斯来说是一个胜利的举动，所以我们返回 true 。

 

提示：

1. 2 <= piles.length <= 500
2. piles.length 是偶数。
3. 1 <= piles[i] <= 500
4. sum(piles) 是奇数。

思路：动态规划：时间O（N^2），空间O（N^2）

• dp[i][j]表示：表示在piles中下标 i 至下标 j 之间的玩家1 所拿石子总数和 玩家2所拿石子总数之差。dp[i][j] > 0 表明玩家1赢，输出True
• 子问题：dp[ i+1 ][ j ]和dp[ i ][ j-1 ]
• 初始状态：dp的初始状态是 i = j 即只有一个石子堆，由于玩家1先拿，则 dp[ i ][ j ] = piles[ i ]
• 状态方程：

dp[ i ][ j ] = max( piles[ i ] - dp[ i+1 ][ j ], piles[ j ] - dp[ i ][ j-1 ])

【解释：当玩家1拿了最左边下标为 i 的石子后，玩家1和玩家2的石子数之差为 piles[ i ] - dp[ i+1 ][ j ], 当玩家1拿了最右边下标为j的那堆石子后，玩家1和玩家2的石子数之差为 piles[ j ] - dp[ i ][ j-1 ], 取其中较大者为新的最优解。】

 

代码：

def stronegame(piles):
    n = len(piles)
    dp = [[0] * n for i in range(n)]
    for i in range(n):
        dp[i][i] = piles[i]
    for i in range(n-2,-1,-1):
        for j in range(i+1,n):
            dp[i][j] = max(piles[i]-dp[i+1][j],piles[j]-dp[i][j-1])
    return dp[0][-1] > 0
piles = [5,3,4,5]
stronegame(piles)

 

转载于:https://www.cnblogs.com/Lee-yl/p/9984090.html