15、合并石头的最低成本

题目描述:
有 N 堆石头排成一排,第 i 堆中有 stones[i] 块石头。

每次移动(move)需要将连续的 K 堆石头合并为一堆,而这个移动的成本为这 K 堆石头的总数。

找出把所有石头合并成一堆的最低成本。如果不可能,返回 -1 。

示例 1:

输入:stones = [3,2,4,1], K = 2
输出:20
解释:
从 [3, 2, 4, 1] 开始。
合并 [3, 2],成本为 5,剩下 [5, 4, 1]。
合并 [4, 1],成本为 5,剩下 [5, 5]。
合并 [5, 5],成本为 10,剩下 [10]。
总成本 20,这是可能的最小值。
示例 2:

输入:stones = [3,2,4,1], K = 3
输出:-1
解释:任何合并操作后,都会剩下 2 堆,我们无法再进行合并。所以这项任务是不可能完成的。.
示例 3:

输入:stones = [3,5,1,2,6], K = 3
输出:25
解释:
从 [3, 5, 1, 2, 6] 开始。
合并 [5, 1, 2],成本为 8,剩下 [3, 8, 6]。
合并 [3, 8, 6],成本为 17,剩下 [17]。
总成本 25,这是可能的最小值。

提示:

1 <= stones.length <= 30
2 <= K <= 30
1 <= stones[i] <= 100

没怎么看懂这个dp
代码:

class Solution {
    public int mergeStones(int[] stones, int K) {
        int l = stones.length;
        if((l-1)%(K-1)>0) 
            return -1;
        
        int[] sum = new int[l+1];
        for(int i=0;i<l;i++){
            sum[i+1] = sum[i] + stones[i];
        }
        
        int[][] dp = new int[l][l];
        for(int step = K; step<=l; step++){
            for(int start=0;start<= l-step; start++){
                int end = step + start -1;
                dp[start][end] = Integer.MAX_VALUE;
                for(int mid=start; mid<end; mid += K-1){
                    dp[start][end] = Math.min(dp[start][end], dp[start][mid]+dp[mid+1][end]);
                }
                if((end-start)%(K-1)==0){
                    dp[start][end] += sum[end+1]-sum[start];
                }
            }
        }
        return dp[0][l-1];
    }
}

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值