Description
在一个圆形操场的四周摆放 N 堆石子,现要将石子有次序地合并成一堆,规定每次只能选相邻的 2 堆合并成新的一堆,并将新的一堆的石子数,记为该次合并的得分。
试设计出一个算法,计算出将 N 堆石子合并成 1 堆的最小得分和最大得分。
Input
数据的第 1 行是正整数 N,表示有 N 堆石子。
第 2行有 N 个整数,第 i 个整数 ai 表示第 i 堆石子的个数。
Output
输出共 2 行,第 1 行为最小得分,第 2 行为最大得分。
经典区间dp
我们观察发现一堆合并后的石子肯定是由若干堆相邻的石子合并而成,我们可以用f[i][j]表示合并完i到j堆石子的代价。若i<k<j,f[i][j]=min(f[i][j],f[i][k]+f[k+1][j]+sum(i,j)),其中sum(i,j)表示第i堆到第j堆的石子总和,但是我们如何保证在求解f[i][j]前f[i][k],f[k+1][j]已知呢,我们可以发现总是先要求解区间长度小的,所以我们就从小到大枚举区间长度len便可以解决。
然而这道题是环状的,于是我们就可以多复制一遍数组接到后面来开环。
最大值同理。
#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=220;
int n ,a[maxn],sum[maxn],f[maxn][maxn],f1[maxn][maxn];
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];a[i+n]=a[i];
sum[i]=sum[i-1]+a[i];
}
for(int i=n+1;i<=2*n;i++)sum[i]=sum[i-1]+a[i];
memset(f,INF,sizeof(f));
for(int len=1;len<=n;len++)
{
for(int i=1;i+len-1<=2*n;i++)
{
int j=i+len-1;
if(len==1)
f[i][j]=0;
else{
for(int k=i;k<j;k++){
f[i][j]=min(f[i][j],f[i][k]+f[k+1][j]+sum[j]-sum[i-1]);
f1[i][j]=max(f1[i][j],f1[i][k]+f1[k+1][j]+sum[j]-sum[i-1]);
// cout<<i<<' '<<k<<' '<<j<<' '<<f[i][j]<<'\n';
}
}
}
}
int ans1=INF,ans2=0;
for(int i=0;i<=n;i++)ans1=min(f[i+1][n+i],ans1),ans2=max(ans2,f1[i+1][n+i]);
cout<<ans1<<endl<<ans2;
return 0;
}