Leetcode 3500. Minimum Cost to Divide Array Into Subarrays

  • [Leetcode 3500. Minimum Cost to Divide Array Into Subarrays](#Leetcode 3500. Minimum Cost to Divide Array Into Subarrays)
    • [1. 解题思路](#1. 解题思路)
    • [2. 代码实现](#2. 代码实现)

1. 解题思路

这一题非常惭愧,没有自己搞定,基本是抄的大佬们的代码,甚至抄完之后还是没完全理解,非常惭愧......

整体这一题的思路还是比较简单的,就是一个动态规划,剩下的问题就在于怎么写这个迭代式。

如果单纯按照题意,设置迭代方式为考察每一个位置作为第i个子串的开头,并考察其子串的终点位置时,其对应的算法复杂度就成了 O ( N 3 ) O(N^3) O(N3),这显然太大了。

因此,我们就必须要调整我们的迭代方式,将其压缩到 O ( N 2 ) O(N^2) O(N2)左右才行。

大佬们的答案最终给出的迭代关系式为如下:
d p ( t , i + 1 ) = min ⁡ j = 0 j = i ( d p ( t − 1 , j ) + ∑ α = 0 i n α ⋅ ∑ β = j + 1 i + 1 c β + k ⋅ ∑ γ = j + 1 N c γ ) dp(t, i+1) = \min\limits_{j=0}^{j=i} (dp(t-1, j) + \sum\limits_{\alpha=0}^{i}n_{\alpha} \cdot \sum\limits_{\beta=j+1}^{i+1}c_{\beta} + k \cdot \sum\limits_{\gamma=j+1}^{N}c_{\gamma}) dp(t,i+1)=j=0minj=i(dp(t−1,j)+α=0∑inα⋅β=j+1∑i+1cβ+k⋅γ=j+1∑Ncγ)

其中, d p ( t , i ) dp(t, i) dp(t,i)表示将前 i i i个数组拆分为至多 t t t个子序列,然后后续 i + 1 i+1 i+1到 n n n个元素作为剩下的子序列时其所需的最小cost。

2. 代码实现

我们将其翻译为最终的python代码语言为:

python 复制代码
class Solution:
    def minimumCost(self, nums: List[int], cost: List[int], k: int) -> int:
        n = len(nums)
        sn = list(accumulate(nums))
        sc = list(accumulate(cost, initial=0))
        
        dp = [math.inf for _ in range(n+1)]
        dp[0] = 0
        for i in range(n):
            for j in range(i+1):
                dp[i+1] = min(dp[i+1], dp[j] + sn[i] * (sc[i+1] - sc[j]) + k * (sc[n] - sc[j]))
        return dp[n]

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

相关推荐
hn小菜鸡11 小时前
LeetCode 377.组合总和IV
数据结构·算法·leetcode
Deepoch11 小时前
Deepoc 大模型:无人机行业的智能变革引擎
人工智能·科技·算法·ai·动态规划·无人机
亮亮爱刷题10 天前
飞往大厂梦之算法提升-7
数据结构·算法·leetcode·动态规划
zmuy10 天前
124. 二叉树中的最大路径和
数据结构·算法·leetcode
chao_78910 天前
滑动窗口题解——找到字符串中所有字母异位词【LeetCode】
数据结构·算法·leetcode
Alfred king10 天前
面试150跳跃游戏
python·leetcode·游戏·贪心算法
芜湖xin10 天前
【题解-Acwing】1022. 宠物小精灵之收服
算法·动态规划·二维费用的背包
呆呆的小鳄鱼10 天前
leetcode:746. 使用最小花费爬楼梯
算法·leetcode·职场和发展
YuTaoShao10 天前
【LeetCode 热题 100】42. 接雨水——(解法一)前后缀分解
java·算法·leetcode·职场和发展
YuforiaCode10 天前
(LeetCode 面试经典 150 题) 27.移除元素
算法·leetcode·面试