LeetCode 题解工作台

石子游戏 VII

石子游戏中,爱丽丝和鲍勃轮流进行自己的回合, 爱丽丝先开始 。 有 n 块石子排成一排。每个玩家的回合中,可以从行中 移除 最左边的石头或最右边的石头,并获得与该行中剩余石头值之 和 相等的得分。当没有石头可移除时,得分较高者获胜。 鲍勃发现他总是输掉游戏(可怜的鲍勃,他总是输),所以他决定尽力 减…

category

4

题型

code_blocks

5

代码语言

hub

3

相关题

当前训练重点

中等 · 状态·转移·动态规划

bolt

答案摘要

我们先预处理出前缀和数组 ,其中 表示前 个石头的总和。 接下来,设计一个函数 $dfs(i, j)$,表示当剩下的石子为 $stones[i], stones[i + 1], \dots, stones[j]$ 时,先手与后手的得分差值。那么答案即为 $dfs(0, n - 1)$。

Interview AiBox logo

Interview AiBox 实时 AI 助手,陪你讲清 状态·转移·动态规划 题型思路

试试 AiBox 面试助手arrow_forward
description

题目描述

石子游戏中,爱丽丝和鲍勃轮流进行自己的回合,爱丽丝先开始

n 块石子排成一排。每个玩家的回合中,可以从行中 移除 最左边的石头或最右边的石头,并获得与该行中剩余石头值之 相等的得分。当没有石头可移除时,得分较高者获胜。

鲍勃发现他总是输掉游戏(可怜的鲍勃,他总是输),所以他决定尽力 减小得分的差值 。爱丽丝的目标是最大限度地 扩大得分的差值

给你一个整数数组 stones ,其中 stones[i] 表示 从左边开始 的第 i 个石头的值,如果爱丽丝和鲍勃都 发挥出最佳水平 ,请返回他们 得分的差值

 

示例 1:

输入:stones = [5,3,1,4,2]
输出:6
解释:
- 爱丽丝移除 2 ,得分 5 + 3 + 1 + 4 = 13 。游戏情况:爱丽丝 = 13 ,鲍勃 = 0 ,石子 = [5,3,1,4] 。
- 鲍勃移除 5 ,得分 3 + 1 + 4 = 8 。游戏情况:爱丽丝 = 13 ,鲍勃 = 8 ,石子 = [3,1,4] 。
- 爱丽丝移除 3 ,得分 1 + 4 = 5 。游戏情况:爱丽丝 = 18 ,鲍勃 = 8 ,石子 = [1,4] 。
- 鲍勃移除 1 ,得分 4 。游戏情况:爱丽丝 = 18 ,鲍勃 = 12 ,石子 = [4] 。
- 爱丽丝移除 4 ,得分 0 。游戏情况:爱丽丝 = 18 ,鲍勃 = 12 ,石子 = [] 。
得分的差值 18 - 12 = 6 。

示例 2:

输入:stones = [7,90,5,1,100,10,10,2]
输出:122

 

提示:

  • n == stones.length
  • 2 <= n <= 1000
  • 1 <= stones[i] <= 1000
lightbulb

解题思路

方法一:记忆化搜索

我们先预处理出前缀和数组 ss,其中 s[i]s[i] 表示前 ii 个石头的总和。

接下来,设计一个函数 dfs(i,j)dfs(i, j),表示当剩下的石子为 stones[i],stones[i+1],,stones[j]stones[i], stones[i + 1], \dots, stones[j] 时,先手与后手的得分差值。那么答案即为 dfs(0,n1)dfs(0, n - 1)

函数 dfs(i,j)dfs(i, j) 的计算过程如下:

  • 如果 i>ji \gt j,说明当前没有石子,返回 00
  • 否则,先手有两种选择,分别是移除 stones[i]stones[i]stones[j]stones[j],然后计算得分差值,即 a=s[j+1]s[i+1]dfs(i+1,j)a = s[j + 1] - s[i + 1] - dfs(i + 1, j)b=s[j]s[i]dfs(i,j1)b = s[j] - s[i] - dfs(i, j - 1),我们取两者中的较大值作为 dfs(i,j)dfs(i, j) 的返回值。

过程中,我们使用记忆化搜索,即使用数组 ff 记录函数 dfs(i,j)dfs(i, j) 的返回值,避免重复计算。

时间复杂度 O(n2)O(n^2),空间复杂度 O(n2)O(n^2)。其中 nn 为石子的数量。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
class Solution:
    def stoneGameVII(self, stones: List[int]) -> int:
        @cache
        def dfs(i: int, j: int) -> int:
            if i > j:
                return 0
            a = s[j + 1] - s[i + 1] - dfs(i + 1, j)
            b = s[j] - s[i] - dfs(i, j - 1)
            return max(a, b)

        s = list(accumulate(stones, initial=0))
        ans = dfs(0, len(stones) - 1)
        dfs.cache_clear()
        return ans
speed

复杂度分析

指标
时间complexity is O(n^2) due to filling an n x n DP table and computing transitions in constant time using prefix sums. Space complexity is O(n^2) for the DP table, reducible to O(n) with optimization since only adjacent subarray results are needed at each step.
空间Depends on the final approach
psychology

面试官常问的追问

外企场景
  • question_mark

    Candidate immediately considers DP subarray state transitions.

  • question_mark

    Candidate tries a naive greedy approach and needs redirection to subarray sums.

  • question_mark

    Candidate correctly identifies that prefix sums optimize repeated sum computations.

warning

常见陷阱

外企场景
  • error

    Forgetting to subtract the removed stone from the remaining sum when computing points.

  • error

    Incorrectly defining dp state leading to miscounted score differences.

  • error

    Overlooking base cases for single-stone subarrays.

swap_horiz

进阶变体

外企场景
  • arrow_right_alt

    Allow players to remove up to k stones from either end per turn.

  • arrow_right_alt

    Modify scoring so removed stone values are added instead of remaining sums.

  • arrow_right_alt

    Extend to three players with cyclic turns and maximize score difference for first player.

help

常见问题

外企场景

石子游戏 VII题解:状态·转移·动态规划 | LeetCode #1690 中等