石子合并

石子合并

题目描述

---

核心思路

这题可以考虑用贪心算法嘛？答案是不能的，因为题目有个限制性条件：只能合并相邻的两堆，而不能随便合并任意位置的两堆石子。如下图所示，采用贪心算法得到的结果就是错误的。

这种题一般都是用区间DP来解决。

• 状态表示：f[i][j]表示将$i$到$j$合并成一堆的方案的集合，属性是最小值min
• 状态计算：
  • 当$i<j$时，$f[i][j]=min(f[i][k]+f[k+1][j]+s[j]-s[i-1])$，其中$i\le k<j$
  • 当$i==j$时，$f[i][j]=0$，因为只有一堆石子，并不需要合并，那么花费就是0 或者说 合并一堆石子的花为0
• 问题答案：f[1][n]表示从第1堆石子合并到第$n$堆石子的最小代价，因此求出了f[1][n]就是答案。

区间 DP 常用模版：

所有的区间dp问题，第一维都是枚举区间长度，一般 len = 1 用来初始化，枚举从 len = 2 开始，第二维枚举左端点(起点)$i$，根据区间的左端点和区间长度，就可以算出区间的右端点了，设区间是$[i,j]$，那么$j-i+1=len$，因此右端点就是$j=len+i-1$。

算法步骤：

• 枚举每个子区间$[i,j]$，然后枚举该区间中的某个分割点$k$，把该区间切分成$[i,k]$和$[k+1,j]$两部分
• 再把这两部分$[i,k]$和$[k+1,j]$合并再一起。设合并左边部分$[i,k]$所花费的最小代价为$f[i][k]$，设得到的新堆为$A$；合并右边部分$[k+1,j]$所花费的最小代价为$f[k+1][j]$，设得到的新堆为$B$，那么合并新堆$A$和$B$的代价就是之前子区间$[i,j]$中这个石子的权值之和，可以用前缀和求出来，也就是$s[j]-s[i-1]$，因此对于合并区间$[i,j]$得到的答案就是：合并左堆所花费的最小代价+合并右堆所花费的最小代价+把这两个新堆合并的代价，即f[i][k]+f[k+1][j]+s[j]-s[i-1]。

时间复杂度：

• 最外面两层循环根据区间长度和区间起点枚举所有子区间，为$O(n^2)$
• 最内层循环枚举这个子区间中的左右分割点，为$O(n)$
• 因此，时间复杂度为$O(n^3)$

以输入1、3、4、2、5为例子：

总结：

---

代码

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=310;
int n;      //石子堆数
int a[N];   //记录每堆石子的质量
int s[N];   //记录前缀和
int f[N][N];    //f[l][r]表示把从l到r合并成一堆的最小代价
int main()
{
    memset(f,0x3f,sizeof f);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);      //输入每堆石子的质量
        s[i]=s[i-1]+a[i];               //计算前缀和
        f[i][i]=0;                      //合并一堆石子的最小代价为0
    }
    //状态计算
    for(int len=2;len<=n;len++) //阶段：枚举区间长度
    {
        for(int l=1;l+len-1<=n;l++) //状态：枚举区间起点
        {
            int r=l+len-1;  //根据区间的起点和区间长度，就可以计算出区间的终点了
            for(int k=l;k<r;k++)      //决策：枚举该子区间[l,r]中的分割点
            {
                f[l][r]=min(f[l][r],f[l][k]+f[k+1][r]+s[r]-s[l-1]);//状态转移方程
            }
        }
    }
    //输出从第1堆合并到第n堆所花费的最小代价f[1][n]
    printf("%d\n",f[1][n]);
    return 0;
}

---