西工大算法实验--石子合并

题目描述

在一个圆形操场的四周摆放着n堆石子(n<= 100)，现要将石子有次序地合并成一堆。规定每次只能选取相邻的两堆合并成新的一堆,并将新的一堆的石子数,记为该次合并的得分。编一程序，读入石子堆数n及每堆的石子数(<=20)。选择一种合并石子的方案,使得做n－1次合并,得分的总和最小； 比如有4堆石子：4 4 5 9 则最佳合并方案如下：
4 4 5 9 score: 0
8 5 9 score: 8
13 9 score: 8 + 13 = 21
22 score: 8 + 13 + 22 = 43

输入
可能有多组测试数据。 当输入n=0时结束! 第一行为石子堆数n(1<=n<=100)； 第二行为n堆的石子每堆的石子数,每两个数之间用一个空格分隔。

输出
合并的最小得分，每个结果一行。

输入样例
4 4 4 5 9 0

输出样例
43

确定算法

区间dp

首先，发现要在一个首尾相连的环上dp并不方便，于是可以断环为链，假设输入的数是a[1],a[2],……,a[n]，那么使a[n+1]=a[1],a[n+2]=a[2],……,a[2n]=a[n]，会发现n个数的环上的每一段都在2*n个数的链上了，于是就在链上dp等价于在环上dp

dp流程

1、状态

设fx[i][j]表示把从i到j的石子合并为一堆的最小得分

2、转移方程

用i,len,j,k 分别定义左端点，长度，右端点，断点
假设存在k使得 将i到k合为一堆的最小得分(a堆)+将k+1到j合为一堆的最小得分(b堆)+将a堆与b堆合并的得分 < 将i到j合为一堆的最小得分 时，更新fx[i][j]，即fx[i][j]=min(fx[i][k]+fx[k+1][j]+将a堆与b堆合并的得分,fx[i][j])，所以现在唯一的未知量就是将a堆与b堆合并的得分了，观察到，a堆的石子个数使a[i]+a[i+1]+……+a[k]，b堆的石子个数是a[k+1]+a[k+2]+……+a[j]，所以将a堆与b堆合并的得分=a[i]+a[i+1]+……+a[j]，故fx[i][j]=min(fx[i][j],fx[i][k]+fx[k+1[j]+a[i]+a[i+1]+……+a[j])

细节问题

（1）初始化
每次计算前都要对最小得分进行重新赋值 ans_min = INT_MAX
（2）前缀和优化
观察到，每次状态转移时，都要计算a[i]+a[i+1]+……+a[j],有些麻烦，所以我们可以与计算前缀和，则a[i]+a[i+1]+……+a[j]=sum[j]-sum[i-1]
（3）数组大小
无论是存每堆的石子数量的a数组，还是pre，fx数组都要开到200以上

代码

#include<iostream>

using namespace std;

int n;
int a[205];
int fx[205][205];
int pre[205];

int min(int a,int b)
{
	return a<b?a:b;
}

int sum(int i,int j)
{
    return pre[j]-pre[i-1];
}

void dp()
{
	int ans_min = INT_MAX;
	int i,len,j,k; //定义左端点，长度，右端点，断点
    len = (n<<1)-1;
    for(i=1;i<=len;i++)
	{
        pre[i]=pre[i-1]+a[i];
    }
    memset(fx,0x7f,sizeof(fx));
    for(i=0;i<=200;i++) fx[i][i] = 0;
    for(i=2;i<=n;i++)
	{
        for(j=1;j<=len-i+1;j++)
		{
            for(k=j;k<j+i-1;k++)
			{
	            int xiao = fx[j][k] +fx[k+1][j+i-1];
	            if( fx[j][j+i-1] > xiao) 
	                fx[j][j+i-1] = xiao;
            }
            fx[j][j+i-1] += sum(j,j+i-1);
        }
    }
    for(i=1;i<=n;i++)
	{
        if( ans_min > fx[i][i+n-1])
        {
        	ans_min = fx[i][i+n-1];
		}
    }
    cout<<ans_min<<endl;
}

int main()
{
	while(cin>>n&&n)
	{
		int i;
		for(i=1;i<=n;i++)
		{
	        cin>>a[i];
	        a[n+i] = a[i];
	    }
		dp();
	}
    return 0;
}