题目链接:AcWing 282. 石子合并
题目
设有N堆石子排成一排,其编号为1,2,3,…,N。
每堆石子有一定的质量,可以用一个整数来描述,现在要将这N堆石子合并成为一堆。
每次只能合并相邻的两堆,合并的代价为这两堆石子的质量之和,合并后与这两堆石子相邻的石子将和新堆相邻,合并时由于选择的顺序不同,合并的总代价也不相同。
例如有4堆石子分别为 1 3 5 2, 我们可以先合并1、2堆,代价为4,得到4 5 2, 又合并 1,2堆,代价为9,得到9 2 ,再合并得到11,总代价为4+9+11=24;
如果第二步是先合并2,3堆,则代价为7,得到4 7,最后一次合并代价为11,总代价为4+7+11=22。
问题是:找出一种合理的方法,使总的代价最小,输出最小代价。
输入格式
第一行一个数N表示石子的堆数N。
第二行N个数,表示每堆石子的质量(均不超过1000)。
输出格式
输出一个整数,表示最小代价。
数据范围
1≤N≤300
输入样例:
4
1 3 5 2
输出样例:
22
思路:
区间dp,设dp[l][r]代表区间合并[l, r]石子的最小代价。
由于递推时需要快速得到区间和,因此先预处理出前缀和sum[]。
初始化:dp[i][j]都设为无穷,dp[i][i] = 0(单独一个石子不需要合并,也就不需要代价,这个情况会用在单个石子和一大堆合并完的石子进行合并时), dp[i][i+1] = sum[i+1] - sum[i-1]
状态转移方程:dp[i][j] = min(dp[i][k] + dp[k+1][j] + sum[j] - sum[i-1], dp[i][j]);
1、递归写法:
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int inf = 0x3f3f3f;
int n, t,dp[1010][1010], sum[1010];
int DP(int l, int r){
if(l == r)
return 0;
if(l + 1 == r)
return dp[l][r] = sum[r] - sum[l-1];
if(dp[l][r] != inf)
return dp[l][r];
else{
for(int i = l; i < r; ++i) // i从L开始,此时是单个石子和一大堆合并完的石子进行合并的情况,不能漏
dp[l][r] = min(dp[l][r], DP(l, i) + DP(i+1, r) + sum[r] - sum[l-1]);
}
return dp[l][r];
}
int main(){
cin >> n;
for(int i = 1; i <= n; ++i){
cin >> t;
sum[i] = sum[i-1] + t;
}
for(int i = 0; i <= n; ++i)
for(int j = 0; j <= n; ++j)
dp[i][j] = inf;
cout << DP(1, n) << endl;
return 0;
}
2、递推写法:
要注意递推的顺序,根据状态转移方程来看,dp[i][j] = min(dp[i][k] + dp[k+1][j] + sum[j]-sum[i-1], dp[i][j]); k是从i开始枚举的,因此算dp[i][j]需要用到dp[大于i][j],所以i要从大到小遍历。
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int inf = 0x3f3f3f;
int n, t, dp[1010][1010], sum[1010];
int main(){
cin >> n;
for(int i = 1; i <= n; ++i){
cin >> t;
sum[i] = sum[i-1] + t;
}
for(int i = 0; i <= n; ++i) // 初始化
for(int j = 0; j <= n; ++j){
dp[i][j] = inf;
dp[i][i] = 0;
}
for(int i = n-1; i >= 1; --i){ //区间左端点
for(int j = i+1; j <= n; ++j){ //区间右端点
for(int 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] << endl;
return 0;
}