石子归并（区间 dp 模板）

石子归并

有N堆石子排成一排,其中第i堆的石子的重量为Ai，现要将石子有次序地合并成一堆。规定每次只能选相邻的2堆合并成新的一堆，形成的新石子堆的重量以及消耗的体力都是两堆石子的重量之和。
求把全部N堆石子合并成一堆最少需要消耗多少体力。

代码

#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <iostream>

using namespace std;

#define ll long long
#define inf 0x3f3f3f3f
ll a[400], sum[400], dp[400][200];

int main() {
    int n;
    while (scanf("%d", &n) != EOF) {
        for (int i = 1; i <= n; i++) {
            scanf("%d", &a[i]);
            sum[i] = sum[i - 1] + a[i];
            dp[i][i] = 0;
        }
        for (int l = 2; l <= n; l++) {
            for (int i = 1; i + l <= n + 1; i++) {
                int j = i + l - 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[i - 1] + sum[j]);
                }
            }
        }
        printf("%d\n", dp[1][n]);
    }
}