石子合并问题--直线版 | ||||||
| ||||||
Description | ||||||
一条直线上摆放着一行共n堆的石子。现要将石子有序地合并成一堆。规定每次只能选相邻的两堆合并成新的一堆,并将新的一堆石子数记为该次合并的得分。请编辑计算出将n堆石子合并成一堆的最小得分和将n堆石子合并成一堆的最大得分。 | ||||||
Input | ||||||
输入有多组测试数据。 每组第一行为n(n<=100),表示有n堆石子,。 二行为n个用空格隔开的整数,依次表示这n堆石子的石子数量ai(0<ai<=100) | ||||||
Output | ||||||
每组测试数据输出有一行。输出将n堆石子合并成一堆的最小得分和将n堆石子合并成一堆的最大得分。 中间用空格分开。 | ||||||
Sample Input | ||||||
3 1 2 3 | ||||||
Sample Output | ||||||
9 11 |
思路:
1、设定dp【i】【j】表示区间从i到j最优合并解、
2、那么其状态转移方程:dp【i】【j】=min/max(dp【i】【k】+dp【k+1】【j】+sum【i】【j】)sum【i】【j】表示从i到j的和。
Ac代码:
#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
int a[1515];
int dp[105][105];
int sum[105];
int main()
{
int n;
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
sum[i]=sum[i-1]+a[i];
}
memset(dp,0x3f3f3f3f,sizeof(dp));
for(int i=1;i<=n;i++)dp[i][i]=0;
for(int d=1;d<n;d++)
{
for(int i=1;i<=n-d;i++)
{
int j=i+d;
for(int k=i;k<j;k++)
{
dp[i][j]=min(dp[i][j],dp[i][k]+dp[k+1][j]);
}
dp[i][j]+=sum[j]-sum[i-1];
}
}
printf("%d ",dp[1][n]);
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)dp[i][i]=0;
for(int d=1;d<n;d++)
{
for(int i=1;i<=n-d;i++)
{
int j=i+d;
for(int k=i;k<j;k++)
{
dp[i][j]=max(dp[i][j],dp[i][k]+dp[k+1][j]);
}
dp[i][j]+=sum[j]-sum[i-1];
}
}
printf("%d\n",dp[1][n]);
}
}