洛谷 P1880 [NOI1995] 石子合并（区间dp）

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

试设计出一个算法,计算出将 NN 堆石子合并成 11 堆的最小得分和最大得分。

输入格式
数据的第 11 行是正整数 NN，表示有 NN 堆石子。

第 22 行有 NN 个整数，第 ii 个整数 a_ia
i
​
表示第 ii 堆石子的个数。

输出格式
输出共 22 行，第 11 行为最小得分，第 22 行为最大得分。

输入输出样例
输入 #1 复制
4
4 5 9 4
输出 #1 复制
43
54
说明/提示
1\leq N\leq 1001≤N≤100，0\leq a_i\leq 200≤a
i
​
≤20。
分析：
将环化为链，即n*2。然后枚举区间长度，左端点，右端点，分隔点。两队石头合并，得到的分数是两堆石头的总个数。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 210;
const ll INF=1e12;
int h[N],a[N],f1[N][N],f2[N][N];//f1求最大分数，f2求最小分数 
int main()
{
	int n;
	scanf("%d",&n);
	memset(f2,0x3f,sizeof(f2));//初始化 
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		a[i+n]=a[i];//将环化为链 
	}
	for(int i=1;i<=2*n;i++)
	{
		h[i]=h[i-1]+a[i];//前缀和 
		f2[i][i]=0;//只有一堆石头，分数为0 
	}
	for(int l=2;l<=n;l++)//枚举区间长度 
	{
		for(int i=1;i<=2*n-l+1;i++)//枚举区间左端点，注意i不能是[1,n]，而是[1,2*n-l+1] 
		{							//因为下面的状态转移方程会用到f1[i][j]中的i可能会大于n 
			int j=i+l-1;//右端点 
			for(int k=i;k<j;k++)// 枚举分隔的点 
			{
				f1[i][j]=max(f1[i][j],f1[i][k]+f1[k+1][j]+h[j]-h[i-1]);
				f2[i][j]=min(f2[i][j],f2[i][k]+f2[k+1][j]+h[j]-h[i-1]);
			}
		}
	}
	int mi=1e8,mx=0;
	for(int i=1;i<=n;i++)
	{
		if(f1[i][i+n-1]>mx) mx=f1[i][i+n-1];
		if(f2[i][i+n-1]<mi) mi=f2[i][i+n-1];
	}
	printf("%d\n%d",mi,mx);
	return 0;
}

优化

#include<stdio.h>
#include<iostream>
#include<math.h>
#include<string>
#include<string.h>
#include<algorithm>
#include<vector>
#include<map>
using namespace std;
#define ll long long
#define eps 1e-8
#define pi acos(-1)
const int N = 1<<19;
int f1[210][210],f2[210][210],s[210][210],h[210],a[210];
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		a[i+n]=a[i];
	}
	for(int i=1;i<=2*n;i++)
	{
		h[i]=h[i-1]+a[i];
		s[i][i]=i;
	}
	int m=2*n;
	for(int l=2;l<=n;l++)//求最小值 
	{
		for(int i=1;i<=m-l+1;i++)
		{
			int j=i+l-1;
			f1[i][j]=1e8;
			for(int k=s[i][j-1];k<=s[i+1][j];k++)//四边形不等式 
			{
				if(f1[i][j]>f1[i][k]+f1[k+1][j]+h[j]-h[i-1])
				{
					f1[i][j]=f1[i][k]+f1[k+1][j]+h[j]-h[i-1];
					s[i][j]=k;
				}
			}
		}
	}
	for(int l=2;l<=n;l++)//求最大值 
	{
		for(int i=1;i<=m-l+1;i++)
		{
			int j=i+l-1;
			f2[i][j]=max(f2[i+1][j],f2[i][j-1])+h[j]-h[i-1];
		}
	}
	int ma=-1,mi=1e8;
	for(int i=1;i<=n;i++)
	{
		mi=min(f1[i][i+n-1],mi);
		ma=max(f2[i][i+n-1],ma);
	}
	printf("%d\n%d",mi,ma);
	return 0;
}