LC每日一题｜20240608 - 3040. 相同分数的最大操作数目 II

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

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

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

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

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

提示：

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

题目等级：Medium

解题思路

第一眼看到这道题还以为是昨天的进阶，仔细一看差别还是挺大的。

整体来说，分类讨论，对于给定的三种分数（nums[0] + nums[1], nums[nums.size - 2] + nums[nums.size - 2], nums[0] + nums[nums.size - 1]），用dfs一层层的扫就是了。

这题很久之前写过，直接cv了～

AC代码

class Solution {
    val map = HashMap<Int, Array<IntArray>>()
    var max = 0
    fun maxOperations(nums: IntArray): Int {
        if (map[nums[0] + nums[1]] == null) map[nums[0] + nums[1]] = Array(nums.size) { IntArray(nums.size + 2) { -1 } }
        if (map[nums[nums.size - 2] + nums[nums.size - 1]] ==  null) map[nums[nums.size - 2] + nums[nums.size - 1]] = Array(nums.size + 2) { IntArray(nums.size) { -1 } }
        if (map[nums[0] + nums[nums.size - 1]] ==  null) map[nums[0] + nums[nums.size - 1]] = Array(nums.size) { IntArray(nums.size) { -1 } }
        dfs(nums, 0, nums.size - 1, nums[0] + nums[1])
        dfs(nums, 0, nums.size - 1, nums[0] + nums[nums.size - 1])
        dfs(nums, 0, nums.size - 1, nums[nums.size - 1] + nums[nums.size - 2])
        return max
    }   

    fun dfs(nums: IntArray, start: Int, end: Int, target: Int): Int {
        var res = 0
        if (start >= end) return res
        if (target == nums[start] + nums[start + 1] && start < nums.size - 2) {
            res = Math.max(res, (if(map[target]!![start + 2][end] == -1) dfs(nums, start + 2, end, target) else map[target]!![start + 2][end]) + 1)
        }

        if (target == nums[start] + nums[end] && start < nums.size - 1 && end >= 1) {
            res = Math.max(res, (if (map[target]!![start + 1][end - 1] == -1) dfs(nums, start + 1, end - 1, target) else map[target]!![start + 1][end - 1]) + 1)
        }

        if (target == nums[end - 1] + nums[end] && end >= 2) {
            res = Math.max(res, (if (map[target]!![start][end - 2] == -1) dfs(nums, start, end - 2, target) else map[target]!![start][end - 2]) + 1)
        }
        map[target]!![start][end] = Math.max(map[target]!![start][end], res)
        max = Math.max(max, map[target]!![start][end])
        return res
    } 
}