石子合并(区间DP)

题目描述

设有 N N N 堆石子排成一排,其编号为 1,2,3,…, N N N

每堆石子有一定的质量,可以用一个整数来描述,现在要将这 N 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 表示石子的堆数 N N N

第二行 N N N 个数,表示每堆石子的质量(均不超过 1000)。

输出格式

输出一个整数,表示最小代价。

数据范围

1≤ N N N≤300

输入样例:

4
1 3 5 2

输出样例:

22

首先,这是一道典型的区间DP问题。

既然是DP首先需要考虑的状态表示,对于该问题,其状态表示为f[i][j]:将区间i~j的石子合并起来的最小代价。

然后再考虑状态转移,由于每次新的一堆石子只能由左右相邻的两堆合并而来,那么对于[i~j]的石子堆,它只能由左边的石子堆f[i][k]以及右边的石子堆f[k + 1][j]合并而来,而这个左、右堆的范围具体划定为多大不清楚,所以需要枚举,那么由此得到了该问题的状态转移方程为:

for k in (i, j) : f [ i ] [ j ] = m i n ( f [ i ] [ j ] , f [ i ] [ k ] + f [ k + 1 ] [ j ] + i ~ j 这段区间的石子重量和 ) f[i][j] = min(f[i][j], f[i][k] + f[k + 1][j] + i~j这段区间的石子重量和) f[i][j]=min(f[i][j],f[i][k]+f[k+1][j]+ij这段区间的石子重量和)

而为了方便快速地计算i~j这段区间的石子重量和,考虑使用前缀和数组s,从而由s[j] - s[i - 1]即可得出,在读数据
时便维护前缀和。

最终,输出f[1][n]即为本问题的全局解。


C++代码

#include <iostream>
#include <algorithm>
using namespace std;

const int N = 310;

int s[N], f[N][N];      //f[i][j]表示范围石子编号i~j中合并石子的最小代价
int n;

int main(){
    cin >> n;
    for(int i = 1;i <= n;i ++){     //计算前缀和,方便后面计算i~j内石子的合并代价
        int t;
        cin >> t;
        s[i] = t + s[i - 1];
    }
    
    //len为2、3、4...n的窗口[i,j]慢慢滑动来维护f[][]
    for(int len = 2;len <= n;len ++)        
        for(int i = 1;i + len - 1 <= n;i ++){
            int j = i + len - 1;
            
            f[i][j] = 0x3f3f3f;
            for(int k = i;k < j;k ++)       
                /**
                * 将i~k、k + 1~j这两堆合并起来的代价是s[j] - s[i - 1]
                * f[i][k] + f[k + 1][j]是前面分别在合并这两堆时积累的代价
                */
                f[i][j] = min(f[i][j], f[i][k] + f[k + 1][j] + s[j] - s[i - 1]);
        }
    
    cout << f[1][n] << endl;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值