基准时间限制:1 秒 空间限制:131072 KB 分值: 20
难度:3级算法题
收藏
关注
N堆石子摆成一条线。现要将石子有次序地合并成一堆。规定每次只能选相邻的2堆石子合并成新的一堆,并将新的一堆石子数记为该次合并的代价。计算将N堆石子合并成一堆的最小代价。
例如: 1 2 3 4,有不少合并方法
1 2 3 4 => 3 3 4(3) => 6 4(9) => 10(19)
1 2 3 4 => 1 5 4(5) => 1 9(14) => 10(24)
1 2 3 4 => 1 2 7(7) => 3 7(10) => 10(20)
括号里面为总代价可以看出,第一种方法的代价最低,现在给出n堆石子的数量,计算最小合并代价。
Input
第1行:N(2 <= N <= 100)
第2 - N + 1:N堆石子的数量(1 <= A[i] <= 10000)
Output
输出最小合并代价
Input示例
4
1
2
3
4
Output示例
19
思路:
dp[i][j]数组记录从第i堆石子到第j堆石子的最小合并代价。
当n=1时,dp[1][1] = 0;
当n=2时,dp[1][2] = a[1]+a[2];
当n=3时,dp[1][3] = min(dp[1][1] + dp[2][3], dp[1][2] + dp[3][3]) + sum[3];
... ...
所以状态方程为:
dp[i][j] = 0 i = j
dp[i][j] = min(dp[i][j], dp[i][k] + dp[k+1][j] + sum[j] - sum[i-1]) i ≠ j
代码:
#include<iostream>
using namespace std;
const int maxn = 100 + 5;
const int INF = 1e9;
int a[maxn], sum[maxn], dp[maxn][maxn];
int n;
int conque(int n) {
for(int i = 0; i < n; i++)
dp[i][i] = 0;
for(int v = 1; v < n; v++) {//每v堆进行一次合并
for(int i = 0; i < n-v; i++) {
int j = i + v;
dp[i][j] = INF;
int tmp = sum[j];
if(i > 0) tmp = sum[j] - sum[i-1];
for(int k = i; k < j; k++) {
dp[i][j] = min(dp[i][j], dp[i][k] + dp[k+1][j] + tmp);
}
}
}
return dp[0][n-1];
}
int main() {
ios::sync_with_stdio(false);
cin >> n;
for(int i = 0; i < n; i++)
cin >> a[i];
sum[0] = a[0];
for(int i = 1; i < n; i++)
sum[i] = sum[i-1] + a[i];
cout << conque(n) << endl;
return 0;
}