Leetcode 3469. Find Minimum Cost to Remove Array Elements

1. 解题思路

这一题我没啥特别好的思路,就只能动态规划了,倒是也能过,不过总是有点勉强……

2. 代码实现

给出python代码实现如下:

class Solution:
    def minCost(self, nums: List[int]) -> int:
        n = len(nums)
        if len(set(nums)) == 1:
            return nums[0] * ((n+1) // 2)
        
        @lru_cache(10**4)
        def dp(pre, idx):
            if idx == n-1:
                return max(nums[pre], nums[idx])
            elif idx == n-2:
                candidates = sorted([nums[pre], nums[idx], nums[idx+1]])
                return candidates[0] + candidates[-1]
            return min(
                max(nums[idx], nums[idx+1]) + dp(pre, idx+2),
                max(nums[pre], nums[idx+1]) + dp(idx, idx+2),
                max(nums[pre], nums[idx]) + dp(idx+1, idx+2)
            )
        
        return dp(0, 1)

提交代码评测得到:耗时6236ms,占用内存344.7MB。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值