文章目錄

• • 1. 題目
  • 2. 解題

1. 題目

有 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

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/minimum-cost-to-merge-stones
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

• dp[i][j] 表示區間 [i, j] 盡量合并后的最小花費，區間可能不能合并為1堆
• 注意枚舉區間中點 mid 時，mid 增量 為 K-1

class Solution { public:int mergeStones(vector<int>& stones, int K) {int n = stones.size();if((n-1)%(K-1) != 0) return -1;//不可能合并vector<vector<int>> dp(n+1, vector<int>(n+1, 0));vector<int> sum(n+1, 0);for(int i = 1; i <= n; i++)sum[i] = stones[i-1] + sum[i-1];//前綴和for(int len = 1; len <= n; len++){ //區間長度for(int i = 1; i+len <= n; i++){ //左端點int j = i+len;//右端點dp[i][j] = INT_MAX;for(int mid = i; mid < j; mid += K-1)//枚舉中間分割點{ //注意mid不能++, 而是 + K-1 , ++的話，很可能左右區間不能合并，dp初始為0，得出的結果小dp[i][j] = min(dp[i][j], dp[i][mid]+dp[mid+1][j]);}if(len%(K-1) == 0)//如果區間能夠合并，加上區間石頭數量dp[i][j] += sum[j]-sum[i-1];}}return dp[1][n];} };

8 ms 8.4 MB C++

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的LeetCode 1000. 合并石头的最低成本（区间DP）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。