Leetcode 3366. Minimum Array Sum

• Leetcode 3366. Minimum Array Sum
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3366. Minimum Array Sum

1. 解题思路

这一题思路上倒是非常的直接，就是一个动态规划，我们考察每一个元素上面是否要进行操作，进行几个操作，然后考察一下最终的结果即可。

2. 代码实现

给出python代码实现如下：

class Solution:
    def minArraySum(self, nums: List[int], k: int, op1: int, op2: int) -> int:
        n = len(nums)
        
        @lru_cache(None)
        def dp(idx, op1, op2):
            if idx >= n:
                return 0
            ans = nums[idx] + dp(idx+1, op1, op2)
            if op1 > 0:
                ans = min(ans, (nums[idx] + 1) // 2 + dp(idx+1, op1-1, op2))
            if op2 > 0 and nums[idx] >= k:
                ans = min(ans, nums[idx] - k + dp(idx+1, op1, op2-1))
            if op1 > 0 and op2 > 0 and nums[idx] >= k:
                if (nums[idx]+1) // 2 >= k:
                    ans = min(ans, min((nums[idx]+1) // 2 - k, (nums[idx]-k+1) // 2) + dp(idx+1, op1-1, op2-1))
                else:
                    ans = min(ans, (nums[idx]-k+1) // 2 + dp(idx+1, op1-1, op2-1))
            return ans
        
        return dp(0, op1, op2)

提交代码评测得到：耗时1704ms，占用内存242.6MB。