AcWing 282. 石子合并(区间dp-两种写法)

题目链接: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;
}
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值