合并石子（c++)

题意

当前有 N 堆石子，他们并列在一排上，每堆石子都有一定的数量。我们需要把这些石子合并成为一堆，每次合并都只能把 相邻 的两堆合并到一起，每一次合并的代价都是这两堆石子的数量之和，经过 N−1 次合并后成为一堆。求把这些石子合并成一堆所需的最小代价。

数据范围

1≤N≤100

解析

通过观察会发现每次合并都是把两个相邻的连续区间进行合并，比如第一次可以把 [1,1] 和 [2,2] 合并成 [1,2][1,2] 。如果之前把 [1,2] 和 [3,4] 合并出来，接着可以把这两段再合并成 [1,4] 。

所以我们可以设 dp[i][j] 表示合并了第 i 堆石子到第 j 堆石子所需的最小代价。

对于当前的最后一次合并操作，是把某两段合并成了 [i,j]，那么可以枚举两段中间的分界点 k，将 [i,j] 拆成 [i,k] 和 [k+1,j]，表示是把这样的两段合并成一段的。

状态转移方程为：

dp[i][j]=min(dp[i][k]+dp[k+1][j]+(a[i]+...+a[j]))

其中 a[i]+...+a[j] 可以用前缀和优化一下，变为 sum[j]−sum[i−1] 。

状态转移方程为：

dp[i][j]=min(dp[i][k]+dp[k+1][j]+(sum[j]−sum[i−1]))

最后答案为 dp[1][N]，因为要枚举 k，所以时间复杂度是 O(n3)。

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int inf = 0x3f3f3f3f;
int a[105], sum[105], dp[105][105];
int main() {
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        sum [i] = sum[i - 1] + a[i];
    }
    for(int len = 2; len <= n; len++){
        for(int i = 1; i + len - 1 <= n; i++){
            int j = i + len - 1;
            dp[i][j] = inf;
            for(int k = i; k < j; k++){
                dp[i][j] = min(dp[i][j], dp[i][k] + dp[k + 1][j] + sum[j] - sum[i - 1]);
            }
        }
    }
    cout << dp[1][n] << endl;
    return 0;
}