LeetCode-Python-3040. 相同分数的最大操作数目 II

给你一个整数数组 nums ，如果 nums 至少 包含 2 个元素，你可以执行以下操作中的 任意 一个：

• 选择 nums 中最前面两个元素并且删除它们。
• 选择 nums 中最后两个元素并且删除它们。
• 选择 nums 中第一个和最后一个元素并且删除它们。

一次操作的 分数 是被删除元素的和。

在确保 所有操作分数相同 的前提下，请你求出 最多 能进行多少次操作。

请你返回按照上述要求 最多 可以进行的操作次数。

示例 1：

输入：nums = [3,2,1,2,3,4]
输出：3
解释：我们执行以下操作：
- 删除前两个元素，分数为 3 + 2 = 5 ，nums = [1,2,3,4] 。
- 删除第一个元素和最后一个元素，分数为 1 + 4 = 5 ，nums = [2,3] 。
- 删除第一个元素和最后一个元素，分数为 2 + 3 = 5 ，nums = [] 。
由于 nums 为空，我们无法继续进行任何操作。

示例 2：

输入：nums = [3,2,6,1,4]
输出：2
解释：我们执行以下操作：
- 删除前两个元素，分数为 3 + 2 = 5 ，nums = [6,1,4] 。
- 删除最后两个元素，分数为 1 + 4 = 5 ，nums = [6] 。
至多进行 2 次操作。

提示：

• 2 <= nums.length <= 2000
• 1 <= nums[i] <= 1000

思路：

首先一共最多有三种分数可能，我们可以针对每种可能进行暴力搜索，注意使用 @cache 装饰器来复用结果优化搜索时间。

时间复杂度：O(3 * N^2)，因为每个可能的暴力搜索最多进行O（n^2)次，对应 start 和 end 分别从 0 到 n - 1

空间复杂度：O（N^2)

class Solution:
    def maxOperations(self, nums: List[int]) -> int:
        @cache
        def dfs(start, end, score):
            if end - start < 1: # nums[start:end + 1], len is (end + 1 - start), which should >= 2
                return 0

            f_two = l_two = fl_two = 0
            if nums[start] + nums[start + 1] == score:
                f_two = 1 + dfs(start + 2, end, score)
            if nums[end] + nums[end - 1] == score:
                l_two = 1 + dfs(start, end - 2, score)
            if nums[start] + nums[end] == score:
                fl_two = 1 + dfs(start + 1, end - 1, score)
            return max(f_two, l_two, fl_two)

        # first two
        f_two = dfs(2, len(nums) - 1, nums[0] + nums[1])
        # last two
        l_two = dfs(0, len(nums) - 3, nums[-2] + nums[-1])
        # first + last
        fl_two = dfs(1, len(nums) - 2, nums[0] + nums[-1])
        return 1 + max(f_two, l_two, fl_two)