AcWing 282. 石子合并

题目链接
AcWing 282. 石子合并

$f[i,j]$：合并 $[i,j]$ 堆石子的最小代价。

通过枚举合并时的分界线，取到 $Min$。

时间复杂度：$O(n^3)$

(二维枚举是 $n^2$，枚举分界线是 $n$，总计 $n^3$)

相关知识：

前缀和，区间DP。

代码：

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 310, INF = 0x3f3f3f3f;

int n;
int s[N]; //前缀和数组
int f[N][N]; // 动态更新数组

int main()
{
    cin >> n;
    
    for (int i = 1; i <= n; i ++ )
    {
        cin >> s[i];
        s[i] += s[i - 1]; // 获得前缀和
    }
    
    // 1.枚举合并区间的长度，保证每次更新时，数据是更新过的
    // 2.len = 1时，表示合并区间为1的石子的代价是0，对应全局变量
    for (int len = 2; len <= n; len ++ )
    // 枚举左端点，范围是 [1, n - len + 1]，保证不越界
        for (int i = 1; i <= n - len + 1; i ++ )
        {
            int l = i, r = i + len - 1;
    // 初次比较时，若f[l][r]未初始化，取最小值答案是0，所以需要初始化为INF
            f[l][r] = INF;
    // 枚举合并时的分界线，范围是 [l, r - 1]，保证合并的最少堆数是1
            for (int k = l; k < r; k ++ )
    // s[r] - s[l - 1] -> [l, r]的区间和
                f[l][r] = min(f[l][r], f[l][k] + f[k + 1][r] + s[r] - s[l - 1]);
        }
    
    // 答案就是 合并[1, n]堆石子的最小代价
    cout << f[1][n] << endl;
    
    return 0;
}