题目描述
设有 N ( N ≤ 300 ) N(N \le 300) N(N≤300) 堆石子排成一排,其编号为 1 , 2 , 3 , ⋯ , N 1,2,3,\cdots,N 1,2,3,⋯,N。每堆石子有一定的质量 m i ( m i ≤ 1000 ) m_i\ (m_i \le 1000) mi (mi≤1000)。现在要将这 N N N 堆石子合并成为一堆。每次只能合并相邻的两堆,合并的代价为这两堆石子的质量之和,合并后与这两堆石子相邻的石子将和新堆相邻。合并时由于选择的顺序不同,合并的总代价也不相同。试找出一种合理的方法,使总的代价最小,并输出最小代价。
输入格式
第一行,一个整数 N N N。
第二行, N N N 个整数 m i m_i mi。
输出格式
输出文件仅一个整数,也就是最小代价。
样例 #1
样例输入 #1
4
2 5 3 1
样例输出 #1
22
解题思路
可以发现,最大的石子堆是由两个小的石子堆合并的,两个小的石子堆也是分别由更小的石子堆合并的……直到石子堆不能再分为止。
那么,只需要将大的石子堆拆成两个小的石子堆,枚举中间的值,每次计算合成这两小堆已用代价加上这次合并的代价。
得到状态转移方程:
d
p
[
i
]
[
j
]
=
m
i
n
(
d
p
[
i
]
[
j
]
,
d
p
[
i
]
[
k
]
+
d
p
[
k
+
1
]
[
j
]
+
a
r
r
[
j
]
−
a
r
r
[
i
−
1
]
)
dp[i][j] = min(dp[i][j],dp[i][k]+dp[k+1][j]+arr[j]-arr[i-1])
dp[i][j]=min(dp[i][j],dp[i][k]+dp[k+1][j]+arr[j]−arr[i−1])
至于合并的代价,可以在输入时就使用前缀和处理。
AC代码
//合并石子
#include<bits/stdc++.h>
using namespace std;
const int N = 305;
int arr[N],a[N],dp[N][N];
int main()
{
int n;
cin>>n;
memset(dp,0x3f,sizeof(dp));//因为求最小,所以初始化正无穷
for(int i=1;i<=n;i++)
{
cin>>a[i];
arr[i] = arr[i-1]+a[i];//前缀和求合并代价
dp[i][i] = 0;//自己本身的代价为0
}
for(int len = 2;len <= n;len++)
{
for(int i=1;i+len-1<=n;i++)
{
int j = i + len -1;
for(int k=i;k<j;k++)
{
dp[i][j] = min(dp[i][j],dp[i][k]+dp[k+1][j]+arr[j]-arr[i-1]);
}
}
}
cout<<dp[1][n];
}