今天讲一道c++的动态规划的题目
石子合并
有 𝑛 堆石子排成一排,第 𝑖 堆石子有 𝑎𝑖 颗,每次我们可以选择相邻的两堆石子合并,代价是两堆石子数目的和,现在我们要一直合并这些石子,使得最后只剩下一堆石子,问总代价最少是多少?
输入格式
第一行一个数字 𝑛。
接下来一行 𝑛 个整数 𝑎1,𝑎2,…,𝑎𝑛
输出格式
一行一个整数,表示答案。
样例输入
3
1 2 3
Copy
样例输出
9
Copy
数据规模
对于所有数据,保证 1≤𝑛,𝑎𝑖≤5001≤𝑛,𝑎𝑖≤500。
这道题是一道区间dp,下面我来分享一下我的思路,希望给大家提供一个参考:
我们开一个数组f[i][j],表示把从第𝑖 堆一直到第𝑛堆的石子合并到一堆所需要的最少代价。那么题目让我们求把第一堆到第n堆的石子合并成一堆。
根据题意,合并两堆石子的代价是两堆石子的石子数目和,所以我们可以定义一个sum,每次去遍历这些石子的数量把他们加起来。但是这样遍历的时间复杂度是O(n),每一次都要遍历一遍,太慢了,所以我们可以用前缀和来储存。我们定义一个s数组,在输入的时候把每一堆前面的所有石子堆的石子数加起来,存在s数组里,这样我们每次访问的时间复杂度就降成了O(1)。
现在我们就要想dp的状态转移方程了。由于每一堆石子可以由两堆合成一堆,那么我们可以枚举一个中间点k,k从我们要合并的区间的左端点开始枚举,小于我们要合并的石子的右端点,从这个中间点把这个要合并的区间分成了两半,那么我们合并这一个区间的代价就可以转化为这分别把这两个区间合并为一个石子堆的代价+把这两个石子堆合并为一个石子堆的代价。这个时候刚刚的前缀和就派上用场了。所以我们的状态转移方程就是 f[i][j]=min(f[i][j],f[i][k]+f[k+1][j]+s[j]-s[i-1]) 这个k就是刚刚枚举的中间点 。这样我们这道题就成功一大半了。
还有就是我们dp外面的循环。我们的外层循环可以来for一遍我们每次合并的区间的长度len,内层循环来遍历左端点i。右端点其实不需要遍历,用左端点加上一个区间的长度len,就是我们右端点的位置。
这样我们这道题就做完了,最后输出一个 f[1][n] 就结束了
下面上参考代码👇:
#include<bits/stdc++.h>
using namespace std;
const int N=1005;
const int inf=0x3f;
int f[N][N],a[N];
int s[N];
int n;
int main(){
cin>>n;
memset(f,inf,sizeof(f));
for(int i=1;i<=n;i++){
cin>>a[i];
s[i]=s[i-1]+a[i];
f[i][i]=0;
}
for(int len=1;len<=n;len++){
for(int i=1;i+len<=n;i++){
int j=i+len;
for(int k=i;k<j;k++){
f[i][j]=min(f[i][j],f[i][k]+f[k+1][j]+s[j]-s[i-1]);
}
}
}
cout<<f[1][n];
return 0;
本人第一篇文章,有不足之处请大家指出来,我会及时改正