洛谷P1880 [NOI1995]石子合并

题目描述

在一个圆形操场的四周摆放N堆石子,现要将石子有次序地合并成一堆.规定每次只能选相邻的2堆合并成新的一堆，并将新的一堆的石子数，记为该次合并的得分。

试设计出1个算法,计算出将N堆石子合并成1堆的最小得分和最大得分.

输入输出格式

输入格式：

数据的第1行试正整数N,1≤N≤100,表示有N堆石子.第2行有N个数,分别表示每堆石子的个数.

输出格式：

输出共2行,第1行为最小得分,第2行为最大得分.

输入输出样例

输入样例#1： 复制

4
4 5 9 4

输出样例#1： 复制

43
54

思路

数据处理：

对于环的问题可以在原数组之后接一段原数组到n-1，使数组长度达到2n-1，然后在新数组上dp。

转移方程：

根据动态规划的思想，我们只要求出每两堆石子合并的最小或最大代价，然后再求出每三堆石子合并的最小或代价，并以此类推就能最终求出 n 堆石子合并的最小或最大代价。我们把 dp[i][j] 定义为合并第 i 堆石子到第 j 堆石子所需的最小代价。很容易就能得到 dp[i][j] = min/max(dp[i][k] + dp[k+1][j]) + sum[j] - sum[i - 1]。sum 表示预处理的石子的前缀和。

实现方式：

可以用第一层循环枚举区间长度，第二层循环枚举起始节点，第三层循环枚举中间节点。

优化（留坑）用On3降到On2：

对于最小值，可以用四边形优化（还不会）。

对于最大值，可以直接贪心，因为对于dp[i][j]求最大值时，dp[i][j]=max(dp[i+1][j],dp[i][j-1])+sum[i]-sum[j]，不需要枚举k（可以自己拿a1a2a3a4证明一下）。

代码

#include<bits/stdc++.h>
using namespace std;
const int N=210; const int MAX=100000;
int n,l,ans,a[N],sum[N],dp[N][N];
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    l=2*n-1;
    for(int i=n+1;i<=l;i++) a[i]=a[i-n];
	for(int i=1;i<=l;i++)
		sum[i]=sum[i-1]+a[i]; 
	for(int h=2;h<=n;h++)
		for(int i=1,j;i<=l-h+1;i++){
			j=i+h-1;
			dp[i][j]=MAX;
			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];
		}
    ans=MAX;
    for(int i=1;i<=n;i++)
        ans=min(dp[i][i+n-1],ans);
	printf("%d\n",ans); ans=0;
	memset(dp,0,sizeof(dp));
	for(int h=2;h<=n;h++)
		for(int i=1,j;i<=l-h+1;i++){
			j=i+h-1;
			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];
		}
	for(int i=1;i<=n;i++)
		ans=max(ans,dp[i][i+n-1]);
	printf("%d",ans);
	return 0;
}