石子归并
Time Limit: 1 Sec Memory Limit: 128 MBSubmit: 355 Solved: 147
[ Submit][ Status][ Web Board]
Description
现在有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
题意:中文题不解释
思路:看到题目光速打了个贪心..嗯WA
改成区间DP后A了也无法想到什么样例可以推翻贪心,如果看客能想到什么样例可以推翻贪心的话,望指教。
dp[i][j]=合并区间[i,j]所需的最小代价
那么 dp[i][j]=min(dp[i][j],dp[i][i+k]+dp[i+k+1][j]+sum[j]-sum[i-1]);
sum为前缀和,sum[j]-sum[i-1]表示区间[i,j]的和,因为不管你是怎么合并的总是要加上这个区间的总和的嘛
k是i~j-1的一个值,表示在哪里分成两个区间的,枚举即可。
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
using namespace std;
#define INF 999999999
int dp[105][105],sum[105];
int main()
{
int n,T,m;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
sum[0]=0;
for(int i=1; i<=n; i++)
{
scanf("%d",&m);
sum[i]=sum[i-1]+m;
}
for(int i=1; i<=n; i++)
for(int j=1; j<=n; j++)
{
if(i==j) dp[i][j]=0;
else dp[i][j]=INF;
}
for(int len=1; len<n; len++)
{
for(int i=1; i<=n; i++)
{
if(i+len>n) break;
int j=i+len;
for(int k=0; i+k<j; k++)
dp[i][j]=min(dp[i][j],dp[i][i+k]+dp[i+k+1][j]+sum[j]-sum[i-1]);
}
}
printf("%d\n",dp[1][n]);
}
return 0;
}