F: 石子合并(NOI1995)
时间限制: 1 Sec 内存限制: 128 MB
题目描述
在操场上沿一直线排列着 n堆石子。现要将石子有次序地合并成一堆。规定每次只能选相邻的两堆石子合并成新的一堆, 并将新的一堆石子数记为该次合并的得分。允许在第一次合并前对调一次相邻两堆石子的次序。
计算在上述条件下将n堆石子合并成一堆的最小得分和初次交换的位置。
输入
输入数据共有二行,其中,第1行是石子堆数n≤100;
第2行是顺序排列的各堆石子数(≤20),每两个数之间用空格分隔。
输出
输出合并的最小得分。
样例输入
3
2 5 1
样例输出
11
Solution
石子合并问题是最经典的DP问题。首先它有如下3种题型:
1.有N堆石子,现要将石子有序的合并成一堆,规定如下:每次只能移动任意的2堆石子合并,合并花费为新合成的一堆石子的数量。求将这N堆石子合并成一堆的总花费最小(或最大)。
分析:当然这种情况是最简单的情况,合并的是任意两堆,直接贪心即可,每次选择最小的两堆合并。本问题实际上就是哈夫曼的变形。
2.有N堆石子,现要将石子有序的合并成一堆,规定如下:每次只能移动相邻的2堆石子合并,合并花费为新合成的一堆石子的数量。求将这N堆石子合并成一堆的总花费最小(或最大)。
分析:我们熟悉矩阵连乘,知道矩阵连乘也是每次合并相邻的两个矩阵,那么石子合并可以用矩阵连乘的方式来解决。
设dp[i][j]表示第i到第j堆石子合并的最优值,sum[i][j]表示第i到第j堆石子的总数量。那么就有状态转移公式:
当然这里可以用到强大的平行四边形优化,证明过程看文件
注意:此题可以交换相邻的石子堆,暴力枚举即可。
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=110;
int dp[maxn][maxn],s[maxn][maxn];
int sum[maxn],a[maxn];
int n,ans=0x3f3f3f3f;
void work()
{
sum[0]=0;
memset(dp,0x3f,sizeof(dp));
for (int i=1;i<=n;i++)
{
sum[i]=sum[i-1]+a[i];
dp[i][i]=0;
}
for (int i=1;i<=n-2+1;i++)
{
int j=i+2-1;
dp[i][j]=dp[i][i]+dp[i+1][j]+sum[j]-sum[i-1];
s[i][j]=i;
}
for (int len=3;len<=n;len++)
for (int i=1;i<=n-len+1;i++)
{
int j=i+len-1;
for (int k=s[i][j-1];k<=s[i+1][j];k++)
if (dp[i][j]>=dp[i][k]+dp[k+1][j]+sum[j]-sum[i-1])
{
dp[i][j]=dp[i][k]+dp[k+1][j]+sum[j]-sum[i-1];
s[i][j]=k;
}
}
if (ans>dp[1][n]) ans=dp[1][n];
}
int main()
{
cin>>n;
for (int i=1;i<=n;i++)
cin>>a[i];
work();
for (int l=1;l<=n-1;l++)
{
swap(a[l],a[l+1]);
work();
swap(a[l],a[l+1]);
}
cout<<ans<<endl;
}