石子合并（弱化版）

石子合并（弱化版）

题目描述

设有 $N(N\le 300)$ 堆石子排成一排，其编号为 $1,2,3,\cdots ,N$。每堆石子有一定的质量 $m_i(m_i\le 1000)$。现在要将这 $N$ 堆石子合并成为一堆。每次只能合并相邻的两堆，合并的代价为这两堆石子的质量之和，合并后与这两堆石子相邻的石子将和新堆相邻。合并时由于选择的顺序不同，合并的总代价也不相同。试找出一种合理的方法，使总的代价最小，并输出最小代价。

输入格式

第一行，一个整数 $N$。

第二行，$N$ 个整数 $m_i$。

输出格式

输出文件仅一个整数，也就是最小代价。

样例 #1

样例输入 #1

4
2 5 3 1

样例输出 #1

22

解题思路

此题做法有好几种，其中一种是 dp，还有一种是记忆化搜索。

dp[i][j]表示合并区间[l,r]的最小代价

那么我们枚举 [l,r]间的一个中转点 k

答案就是：
1.不合并；
2.合并，加上代价；

取二者之中最小值。
dp[i][i+l]=min(dp[i][l+i],dp[i][k]+dp[k+1][i+l]-sum[i-1]+sum[i+l]);

sum 数组存的是前缀和，快速求合并 [𝑙,𝑟]的代价。

参考代码

#include<bits/stdc++.h>
using namespace std;
int dp[500][500],sum[1010],a[1010];
int main(){
    int n;
    cin>>n;
    memset(dp,0x3f3f3f,sizeof(dp));//初始化
    for(int i=1;i<=n;i++){
        cin>>a[i];
        sum[i]=sum[i-1]+a[i];//前缀和
        dp[i][i]=0;//初始化d[i][i]
    }
    for(int l=1;l<n;l++){
        for(int i=1;i<=n;i++){
            for(int k=i;k<i+l;k++){
                if(i+l<=n)
                dp[i][i+l]=min(dp[i][l+i],dp[i][k]+dp[k+1][i+l]-sum[i-1]+sum[i+l]);//动态转移方程
            }
        }
    }
    cout<<dp[1][n];//输出
    return 0;
}