题目描述
在一个圆形操场的四周摆放N堆石子,现要将石子有次序地合并成一堆.规定每次只能选相邻的2堆合并成新的一堆,并将新的一堆的石子数,记为该次合并的得分。
试设计出1个算法,计算出将N堆石子合并成1堆的最小得分和最大得分.
输入输出格式
输入格式:数据的第1行试正整数N,1≤N≤100,表示有N堆石子.第2行有N个数,分别表示每堆石子的个数.
输出格式:输出共2行,第1行为最小得分,第2行为最大得分.
输入输出样例
思路
数据处理:
对于环的问题可以在原数组之后接一段原数组到n-1,使数组长度达到2n-1,然后在新数组上dp。
转移方程:
根据动态规划的思想,我们只要求出每两堆石子合并的最小或最大代价,然后再求出每三堆石子合并的最小或代价,并以此类推就能最终求出 n 堆石子合并的最小或最大代价。我们把 dp[i][j] 定义为合并第 i 堆石子到第 j 堆石子所需的最小代价。很容易就能得到 dp[i][j] = min/max(dp[i][k] + dp[k+1][j]) + sum[j] - sum[i - 1]。sum 表示预处理的石子的前缀和。
实现方式:
可以用第一层循环枚举区间长度,第二层循环枚举起始节点,第三层循环枚举中间节点。
优化(留坑)用On3降到On2:
对于最小值,可以用四边形优化(还不会)。
对于最大值,可以直接贪心,因为对于dp[i][j]求最大值时,dp[i][j]=max(dp[i+1][j],dp[i][j-1])+sum[i]-sum[j],不需要枚举k(可以自己拿a1a2a3a4证明一下)。
代码
#include<bits/stdc++.h>
using namespace std;
const int N=210; const int MAX=100000;
int n,l,ans,a[N],sum[N],dp[N][N];
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
l=2*n-1;
for(int i=n+1;i<=l;i++) a[i]=a[i-n];
for(int i=1;i<=l;i++)
sum[i]=sum[i-1]+a[i];
for(int h=2;h<=n;h++)
for(int i=1,j;i<=l-h+1;i++){
j=i+h-1;
dp[i][j]=MAX;
for(int k=i;k<j;k++)
dp[i][j]=min(dp[i][j],dp[i][k]+dp[k+1][j]);
dp[i][j]+=sum[j]-sum[i-1];
}
ans=MAX;
for(int i=1;i<=n;i++)
ans=min(dp[i][i+n-1],ans);
printf("%d\n",ans); ans=0;
memset(dp,0,sizeof(dp));
for(int h=2;h<=n;h++)
for(int i=1,j;i<=l-h+1;i++){
j=i+h-1;
for(int k=i;k<j;k++)
dp[i][j]=max(dp[i][j],dp[i][k]+dp[k+1][j]);
dp[i][j]+=sum[j]-sum[i-1];
}
for(int i=1;i<=n;i++)
ans=max(ans,dp[i][i+n-1]);
printf("%d",ans);
return 0;
}