题目描述
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堆石子的数量,计算最小合并代价。
收起
输入
第1行:N(2 <= N <= 100)
第2 - N + 1:N堆石子的数量(1 <= A[i] <= 10000)
输出
输出最小合并代价
输入样例
4
1
2
3
4
输出样例
19
思路
石子归并每次将一堆石子和另一堆合并在一起,代价为两堆石子的数量总和,我们可以知道到最后将要得出结果时一定是只剩下两堆,然后一和并即是结果,所以我们可以考虑最后两堆是如何得来的,使其最小,便可以使最后的结果最小,所以就是将这一长串分成了一个个区间,求最小的区间,来得到最小的结果,便考虑区间dp
递归定义合并A,B两堆石子的代价:合并A堆的代价+合并B堆的代价+A堆石子重量+B堆石子重量。
区间dp的详解可以看这个视频:https://www.bilibili.com/video/av12159085?from=search&seid=9519642503896451437&tdsourcetag=s_pcqq_aiomsg
AC代码
#include<cstdio>
#include<iostream>
using namespace std;
#include<numeric>
#include<algorithm>
#include<cstring>
const int INF=1e7;
const int maxn=105;
__int64 a[maxn],s[maxn],dp[maxn][maxn];
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%I64d",&a[i]);
partial_sum(a,a+n+1,s); //用stl里的函数求前缀的和,求好后放入s数组中
for(int d=1;d<n;++d)
for(int i=1,j;(j=i+d)<=n;++i)
{
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]+s[j]-s[i-1]);
} s[j]-s[i-1]即i到j的和
printf("%I64d\n",dp[1][n]);
return 0;
}
题目链接:http://www.51nod.com/Challenge/Problem.html#!#problemId=1021