【训练题】石子归并
Description
设有 n 堆石子排成一排,其编号为1、2、3、…、n(n<=100)。每堆石子的数量用:a1、a2、…、an表示,现将这n堆石子归并成一堆,归并的规则:
◆每次只能将相邻两堆归并成一堆,即:第1堆石子a[1]只能与第2堆石子a[2]归并,最后一堆石子a[n]只能与a[n-1]归并,中间的石子a[i]只能与a[i-1]或a[i+1]归并;
◆每次归并的代价是两堆石子的重量之和。
例如:假设有这样7堆石子,各堆石子的数量分别是:13、7、8、16、21、4、18,将它们归并成一堆有多种归并方法,那么当给出了n堆石子的数量之后,请你求出最小的总代价和最大总代价。
Input
第一行一个整数N ,表示有N堆石子; 接下来的N行,每行一个整数,第i+1行表示第i堆石子的数量;
Output
第一行一个整数,表示最小总代价; 第二行一个整数,表示最大总代价;
Sample Input 1
4
7
6
5
6
Sample Output 1
48
55
Hint
n<=100。每堆石子的数量不超过1000。
#include<bits/stdc++.h>
using namespace std;
int a[105];
int sum[105];
int f[105][105];
int main(){
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
sum[i]=sum[i-1]+a[i]; //前缀和的计算
}
memset(f,0x7f,sizeof(f)); //初始化为极大值(0x7f=127)
for(int i=1;i<=n;i++) f[i][i]=0;
for(int i=n-1;i>=1;i--){
for(int j=i+1;j<=n;j++){
for(int k=i;k<=j-1;k++){
f[i][j]=min(f[i][j],f[i][k]+f[k+1][j]+sum[j]-sum[i-1]); //i到k之前合并的值+k+1到j之前合并的值+i—j的值(这次合并的值)
}
}
}
printf("%d\n",f[1][n]);
//
memset(f,-0x7f,sizeof(f)); 初始化为极小值(-0x7f=-127)
for(int i=1;i<=n;i++) f[i][i]=0;
for(int i=n-1;i>=1;i--){
for(int j=i+1;j<=n;j++){
for(int k=i;k<=j-1;k++){
f[i][j]=max(f[i][j],f[i][k]+f[k+1][j]+sum[j]-sum[i-1]);
}
}
}
printf("%d\n",f[1][n]);
return 0;
}