石子合并问题

问题描述

有n堆石子排成一排，每堆石子有一定的数量，先要将n堆石子合并为一堆，合并只能每次将相邻的两堆石子合并为一堆，每次合并花费的代价为这两堆石子的和，经过n-1次合并后成为一堆，求出总代价的最小值。

状态转移方程

其中 表示从i到j花费的最小代价， 表示从i到j的所有石子数量和。

算法思想

使用动态规划的思想。一共有n堆石子，每次合并相邻的两堆石子，合并后得到两堆石子的数量总和。最终合并后得到的数量和最少。
我们设sum（i,j）定义为第i堆石子到第j堆石子合并后得到的总数量，a(i)为第i堆石子的数量。

1. 当合并的石子堆为1堆时，sum（i,i）的值为0。
2. 当合并的石子堆为2堆时，sum（i,i+1）的值为a（i）+a（i+1）
3. 当合并的石子堆为3堆时，sum（i,i+2）的值为
   MIN ((sum(i,i)+sum(i+1,i+2)+sum(i,i+2)),(sum(i,i+1)+sum(i+2,i+2)+sum(i,i+2))
4. 当合并的石子堆为4堆时……

程序实现

#include <iostream>
#include <limits.h>
#include <cstring>
#include <stdio.h>
using namespace std;
const int N=210;//定义最大值
const int INF=0x3f3f3f3f;//定义无穷值
# define min(a,b)a<b?a:b//定义函数，取两个值的较小者
int n;//表示n堆石子
int a[N], dp[N][N], sum[N];
//数组a[]存储每堆石子的数量，用dp[i][j]表示从i到j的最小合并成本， sum[i][j]表示从i到j的所有石子重量和。

int getMinValue(int n){
	memset(dp, 0, sizeof(dp));//初始化dp数组，元素置0
	for(int v=1; v<n; v++){	//v表示i到j的间距
		for(int i=0; i<n-v; i++){//循环每个间距
			int j=i+v;
			int tmp=sum[j]-(i>0? sum[i-1]:0);//tmp存储从i到j的合并成本
			dp[i][j]=INF;
			for(int k=i; k<j; k++){
				dp[i][j]=min(dp[i][j], dp[i][k]+dp[k+1][j]+tmp);//取从i到j的合并代价最小值
			}
		}
	}
	return dp[0][n-1];
}

int main(){
	while(scanf("%d", &n)!=EOF){
		for(int h=0;h<n; h++){
			scanf("%d", &a[h]);//输入每个石子堆的石子数量
		}
		sum[0]=a[0];
		for(int i=1; i<n; i++){
			sum[i]=sum[i-1]+a[i];//填充sum数组
		}
		printf("%d\n", getMinValue(n));
	}
	return 0;
}