CODEVS 1048 石子归并

题目描述 Description

有n堆石子排成一列，每堆石子有一个重量w[i], 每次合并可以合并相邻的两堆石子，一次合并的代价为两堆石子的重量和w[i]+w[i+1]。问安排怎样的合并顺序，能够使得总合并代价达到最小。

输入描述 Input Description

第一行一个整数n（n<=100）

第二行n个整数w1,w2…wn (wi <= 100)

输出描述 Output Description

一个整数表示最小合并代价

样例输入 Sample Input

4

4 1 1 4

样例输出 Sample Output

18

分析

若用fmin[i,j]记录合并从i开始j堆石子的最小值，用sum[i,j]记录从i开始j堆石子重量的和，那么转换方程如下：

if (f[i,k]+f[k+i,j]+sum[i,j])>fmin[i,j]
fmin[i,j]:=f[i,k]+f[k+i,j]+sum[i,j];
从1到j-1枚举k从而求出从i开始j堆石子的合并最小值；

代码

#include <cstdio>
#define MAXN 101
#define INF 0x7fffffff

int stone[MAXN];
int f[MAXN][MAXN];
int sum[MAXN];
int n;

int min (int a, int b) {
    return a < b ? a : b;
}

int main() {

    scanf("%d", &n);

    for (int i = 1; i <= n; i++) {
        scanf("%d", &stone[i]);
        sum[i] = sum[i - 1] + stone[i];
    }

    for (int p = 1; p <= n - 1; p++)
        for (int i = 1; i <= n - p; i++) {

            int j = i + p;

            f[i][j] = INF;

            for (int k = i; k <= j - 1; k++) 
                f[i][j] = min (f[i][j], f[i][k] + f[k + 1][j]);

            f[i][j] = f[i][j] + sum[j] - sum[i - 1];

        }

    printf("%d", f[1][n]);

    return 0;
}