Leetcode 3473. Sum of K Subarrays With Length at Least M

1. 解题思路

这一题我的思路上同样走的是动态规划的思路。

我们考察每一个位置上的字符,它有三种状态:

  1. 作为一个子串的开头位置(此时要求后续至少有m-1个字符,且它们必然也都属于该子串)
  2. 作为上一个长度至少为m的子串的一部分(此时要求前一个字符必然也属于上一个子串)
  3. 不使用该位置上的字符到任何子串当中

我们遍历每一个位置的情况即可得到我们的答案。

2. 代码实现

给出python代码实现如下:

class Solution:
    def maxSum(self, nums: List[int], k: int, m: int) -> int:
        n = len(nums)
        
        @lru_cache(None)
        def dp(idx, k, in_subarray):
            if idx >= n:
                return 0 if k == 0 else -math.inf
            elif k == 0:
                return 0 if not in_subarray else max(nums[idx] + dp(idx+1, k, in_subarray), 0)
            if (n-idx) < m * k:
                return -math.inf
            elif (n-idx) == m * k:
                return sum(nums[idx:])
            
            if not in_subarray:
                return max(sum(nums[idx:idx+m]) + dp(idx+m, k-1, True), dp(idx+1, k, False))
            else:
                return max(nums[idx] + dp(idx+1, k, True), sum(nums[idx:idx+m]) + dp(idx+m, k-1, True), dp(idx+1, k, False))
            
        return dp(0, k, False)

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值