传送门 \looparrowright

题目描述

设有N堆沙子排成一排,其编号为 1,2,3,,N(N300)1,2,3,\dots,N(N\leq 300)。每堆沙子有一定的数量,可以用一个整数来描述,现在要将这 NN 堆沙子合并成为一堆,每次只能合并相邻的两堆,合并的代价为这两堆沙子的数量之和,合并后与这两堆沙子相邻的沙子将和新堆相邻,合并时由于选择的顺序不同,合并的总代价也不相同,如有 44 堆沙子分别为 1,3,5,21, 3 ,5, 2,我们可以先合并 1,21,2 堆,代价为 44 ,得到 4,5,24, 5, 2, 又合并 1,21,2 堆,代价为 99,得到 9,29, 2 ,再合并得到 11,总代价为 4+9+11=244+9+11=24,如果第二步是先合并 2,32,3 堆,则代价为 77,得到 4,74,7,最后一次合并代价为 1111,总代价为 4+7+11=224+7+11=22。问题是:找出一种合理的方法,使总的代价最小。输出最小代价。

输入描述

第一行一个数 NN 表示沙子的堆数 NN
第二行 NN 个数,表示每堆沙子的质量 (1000)(\leq1000)

输出描述

合并的最小代价

示例1

输入
4
1 3 5 2
输出
22

分析

经典区间 DP\text{DP} 问题。
定义 dp[i][j]dp[i][j] 为合并区间 [i,j][i,j] 的最小代价。则有 dp[i][j]=minik<j(dp[i][k]+dp[k+1][j]+count(i,j))dp[i][j]=\min\limits_{i\le k< j} \left(dp[i][k]+dp[k+1][j]+count(i,j)\right)count(i,j)count(i,j) 为区间 [i,j][i,j] 内石子的个数)。相当于将 [i,j][i,j] 拆成两个区间 [i,k][i,k][k+1,j][k+1,j],两个区间各自合并石子需要 dp[i][k]dp[i][k]dp[k+1][j]dp[k+1][j] 的代价,将两个区间合并要花费 count(i,j)count(i,j) 的代价。
进行递推时,最外层循环为 jj,控制右边界;第二层循环从 jj11,控制左边界;第三层循环控制分界点。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#include<algorithm>
#include<cstring>
#include<iostream>
#include<cstdio>
#define N 302
using namespace std;
int dp[N][N];
int n;
int sum[N];
int a[N];
int main()
{
memset(dp,0x3f,sizeof(dp));//初始化
int i,j,k;
cin>>n;
for(i=1;i<=n;i++) scanf("%d",a+i);
//预处理前缀和,O(1)得到count(i,j)
for(i=1;i<=n;i++) sum[i]=sum[i-1]+a[i];
//注意循环的顺序
for(j=1;j<=n;j++)
{
//逆序更新,优先更新靠近边界的值
for(i=j;i>=1;i--)
{
if(i==j) dp[i][j]=0;
for(k=i;k<j;k++)
{
dp[i][j]=min(dp[i][k]+dp[k+1][j]+sum[j]-sum[i-1],dp[i][j]);
}
}
}
cout<<dp[1][n];
return 0;
}