LeetCode题解(1690)：石子游戏VII(Python)

最新推荐文章于 2021-06-16 20:31:13 发布

长行

最新推荐文章于 2021-06-16 20:31:13 发布

阅读量381

点赞数

分类专栏：刷题记录文章标签： leetcode python 动态规划

本文链接：https://blog.csdn.net/Changxing_J/article/details/112138291

版权

刷题记录专栏收录该内容

1966 篇文章 55 订阅

订阅专栏

博客内容介绍了如何使用动态规划解决一个名为石子游戏VII的问题。代码实现了一个Python类，通过计算前缀和并建立动态规划矩阵来确定玩家在剩余一定数量石子时的最佳策略。时间复杂度和空间复杂度均为O(N^2)，并在给定的测试用例中给出了执行时间。

摘要由CSDN通过智能技术生成

题目：原题链接（中等）

标签：动态规划

解法	时间复杂度	空间复杂度	执行用时
Ans 1 (Python)	$O(N^2)$	$O(N^2)$	4052ms (60.65%)
Ans 2 (Python)
Ans 3 (Python)

解法一：

class Solution:
    def stoneGameVII(self, stones: List[int]) -> int:
        n = len(stones)

        # 计算前缀和数列
        prefix = [0]
        now = 0
        for s in stones:
            now += s
            prefix.append(now)

        # print("前缀和:", prefix)

        dp = [[0] * n for _ in range(n)]

        for l in range(1, n):
            for i in range(n - l):
                j = i + l

                # 当剩下2个石头时
                if l == 1:
                    dp[i][j] = max(stones[i], stones[j])

                # 当剩下3个及以上的石头时
                else:
                    # print((i, j), stones[i:j + 1], "拿走:", stones[i], "->", (prefix[j + 1] - prefix[i + 1]),
                    #       dp[i + 1][j], "=", prefix[j + 1] - prefix[i + 1] - dp[i + 1][j])
                    # print((i, j), stones[i:j + 1], "拿走:", stones[j], "->", (prefix[j] - prefix[i]), dp[i][j - 1], "=",
                    #       (prefix[j] - prefix[i]) - dp[i][j - 1])
                    dp[i][j] = max((prefix[j + 1] - prefix[i + 1]) - dp[i + 1][j],  # 拿掉i
                                   (prefix[j] - prefix[i]) - dp[i][j - 1])  # 拿掉j

        # for row in dp:
        #     print(row)

        return dp[0][-1]