现在有n堆石子,第i堆有ai个石子。现在要把这些石子合并成一堆,每次只能合并相邻两个,每次合并的代价是两堆石子的总石子数。求合并所有石子的最小代价。
Input
第一行包含一个整数T(T<=50),表示数据组数。
每组数据第一行包含一个整数n(2<=n<=100),表示石子的堆数。
第二行包含n个正整数ai(ai<=100),表示每堆石子的石子数。
Output
每组数据仅一行,表示最小合并代价。
Sample Input
2 4 1 2 3 4 5 3 5 2 1 4
Sample Output
19 33
注意初始化和前缀和
#include<iostream>
#include<string.h>
using namespace std;
const int INF = 1<<29;
const int MAX = 105;
int A[MAX];
int dp[MAX][MAX]; //dp[i][j]表示合并区间[i,j]内的石头的最小代价
int sum[MAX];
void solve(int n)
{
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
dp[i][j] = INF;
for (int i = 1; i <= n; i++)
dp[i][i]= 0;
for (int l = 2; l <= n; l++) //区间从2开始到n
{
for (int i = 1; i <= n - l + 1; i++) //从左边开始考虑
{
for (int j = i; j <= i + l - 1; j++) //选择合并的两堆
{
dp[i][i+l-1] = min(dp[i][i+l-1], dp[i][j] + dp[j+1][i+l-1] + sum[i+l-1] - sum[i-1]);
}
}
}
}
int main()
{
int T;
cin >> T;
while (T--)
{
int n;
cin >> n;
sum[0] = 0;
for (int i = 1; i <= n; i++)
{
cin >> A[i];
sum[i] = A[i] + sum[i-1];
}
solve(n);
cout << dp[1][n] << endl;
}
return 0;
}