试题 算法提高 合并石子

问题描述

　　在一条直线上有n堆石子，每堆有一定的数量，每次可以将两堆相邻的石子合并，合并后放在两堆的中间位置，合并的费用为两堆石子的总数。求把所有石子合并成一堆的最小花费。

输入格式

　　输入第一行包含一个整数n，表示石子的堆数。
　　接下来一行，包含n个整数，按顺序给出每堆石子的大小 。

输出格式

　　输出一个整数，表示合并的最小花费。

样例输入

5
1 2 3 4 5

样例输出

33

数据规模和约定

　　1<=n<=1000, 每堆石子至少1颗，最多10000颗。

分析：

此问题为区间dp,所以设dp[i][j]为合并此区间的最小花费，选取一个区间任意点k,由此可写出状态转移方程为：dp[i][j]=min(dp[i][j],dp[i][k]+dp[k+1][j]+sum)//sum为合并此区间的花费。

至于sum我们可用前缀和进行优化，使时间复杂度更低。

下面为代码：

#include<iostream>
#include<algorithm>
using namespace std;
int a[1010],dp[1010][1010],n;//区间dp 
int main(){
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);//此为加快速度的
	 cin>>n;
	 for(int i=1;i<=n;i++)cin>>a[i],a[i]+=a[i-1];//前缀和
	 for(int len=2;len<=n;len++)//枚举长度，长度为1无意义，为一堆，无需合并
	 for(int i=1;i+len-1<=n;i++)//枚举左端点
{int j=i+len-1;dp[i][j]=1e9;//右端点,初始化区间最小花费 
for(int k=i;k<j;k++)//枚举K 
dp[i][j]=min(dp[i][j],dp[i][k]+dp[k+1][j]+a[j]-a[i-1]);
 } 
 cout<<dp[1][n];//合并1....n区间的最小花费 
	return 0;
}