时间限制:1s;内存限制:256MB
【问题描述】
还记得经典题石子合并吗?现在小 Y 将题目加强啦!
在一个圆形操场的四周摆放着 n 堆石子,现要将石子有次序地合并成一堆。规定每次只能选取相邻的三堆合并成新的一堆,并将新的一堆的石子数,记为该次合并的得分
编一程序,读入石子堆数 n 及每堆的石子数。选择一种合并石子的方案,使得做(n-1)/2次合并,得分的总和最小
【输入文件】
第 1 行一个数,表示石子堆数
第 2 行是顺序排列的各堆石子数(≤1000),每两个数之间用空格分隔
【输出文件】
输出合并的最小得分
【样例输入】
5
1 2 3 4 5
【样例输出】
21
【样例解释】
先合并(1 2 3),再合并(6 4 5)
【数据范围】
对于 20%的数据,n=5
对于 60%的数据,n<=80
对于 100%的数据,n<=400
把三个合并
拆成合并两个和一个
利用辅助数组
减少一层循环
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<queue>
#include<vector>
#include<climits>
#include<string>
#include<cstdlib>
#include<ctime>
#include<map>
#define LL long long
#define MOD 1000000007
using namespace std;
int n,i,j,k,dp1[1005][1005],s[2005],dp2[1005][1005],sum[1005];
int main()
{
freopen("merge.in","r",stdin);
freopen("merge.out","w",stdout);
scanf("%d",&n);
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
{
dp1[i][j]=1e9;
dp2[i][j]=1e9;
}
for(i=1;i<=n;i++)
{
scanf("%d",&s[i]);
sum[i]=sum[i-1]+s[i];
}
for(i=n;i;i--)
{
dp1[i][i]=0;
dp1[i][i+2]=sum[i+2]-sum[i-1];
for(j=i+3;j<=n;j++)
{
for(k=i;k<j;k++) dp2[i][j]=min(dp2[i][j],dp1[i][k]+dp1[k+1][j]);
for(k=i;k<j;k++) dp1[i][j]=min(dp1[i][j],min(dp1[i][k]+dp2[k+1][j],dp2[i][k]+dp1[k+1][j])+sum[j]-sum[i-1]);
}
}
printf("%d",dp1[1][n]);
return 0;
}