[区间dp]合并石子

问题描述

在一个圆形操场的四周摆放 $n$ 堆石子,现要将石子有次序地合并成一堆.规定每次只能选相邻的 $2$ 堆合并成新的一堆，并将新的一堆的石子数，记为该次合并的得分。试求出计算出将 $n$ 堆石子合并成 $1$ 堆的最小得分和最大得分。

输入格式

第一行输入一个整数 $n$ ，表示有 $n$ 堆石子。
第二行输入 $n$ 个整数，表示第 $i$ 堆石子的个数 $a_i$。

输出格式

输出一行两个整数，表示最小得分和最大得分。

样例

样例输入1：

3
1 2 3

样例输出1：

9 11

数据范围

$1\le n\le 100$
$1\le a_i\le 1000$

题解

当看见合并这类的问题，可以想到 区间dp 和 贪心。由于这道题并不能局部最优推出全局最优，因此使用 区间dp。

定义 $dp{1}_{i,j}$ 表示从 $i$ 到 $j$ 最小得分，$dp{2}_{i,j}$ 表示从 $i$ 到 $j$ 最大得分。

最开始每堆石子都没有合并，因此 $dp{1}_{i,i}=dp{2}_{i,i}=0$。

转移方程以 $dp1$ 为例。
将 $i$ 到 $j$ 合并为一堆石子，可以把 $i$ 到 $k$ 和 $k+1$ 到 $j$ 合并起来（$i\le k<j$），即 $dp{1}_{i,k}+dp{1}_{k+1,j}$，最后加上 $i$ 到 $j$ 的和即可。
$dp{1}_{i,j}=min(dp{1}_{i,j},dp{1}_{i,k}+dp{1}_{k+1,j}+su{m}_j-su{m}_{i-1})$

同理，$dp{2}_{i,j}=max(dp{2}_{i,j},dp{2}_{i,k}+dp{2}_{k+1,j}+su{m}_j-su{m}_{i-1})$。

最后输出 $dp{1}_{1,n}$ 和 $dp{2}_{1,n}$ 即可。

但是由于是圆形，因此需要破环为链。

可将原数组再复制一遍，进行计算。

int s[210];//前缀和
int f1[210][210], f2[210][210];//dp1 和 dp2
int main(){
	输入
	//破环为链（复制）
	for(int i = 1; i < n; ++ i){
		a[i + n] = a[i];
	}
	//前缀和
	//因为 1 到 n 相当于 n + 1 到 2 * n，因此循环到 2 * n - 1即可
	for(int i = 1; i < 2 * n; ++ i){
		s[i] = s[i - 1] + a[i];
	}
	for(int i = 1; i < n; ++ i){
		for(int j = 1; j <= 2 * n - i; ++ j){
			int t = i + j;
			f1[j][t] = 2147483647;
			//合并 j 到 t 的区间
			for(int k = j; k < t; ++ k){
				f1[j][t] = min(f1[j][k] + f1[k + 1][t], f1[j][t]);
			}
			f1[j][t] += s[t] - s[j - 1]; 
		} 
	}
	int ans1 = 2147483647;
	for(int i = 1; i < n; ++ i){
		ans1 = min(ans1, f1[i][i + n - 1]);
		//从 i 到 i + n - 1，相当于从 i 开始合并的数组
	}
	for(int i = 1; i < n; ++ i){
		for(int j = 1; j <= 2 * n - i; ++ j){
			int t = i + j;
			f2[j][t] = -2147483647;
			//合并 j 到 t 的区间
			for(int k = j; k < t; ++ k){
				f2[j][t] = max(f2[j][k] + f2[k + 1][t], f2[j][t]);
			}
			f2[j][t] += s[t] - s[j - 1]; 
		} 
	}
	int ans2 = -2147483647;
	for(int i = 1; i < n; ++ i){
		ans2 = max(ans2, f2[i][i + n - 1]);
		//从 i 到 i + n - 1，相当于从 i 开始合并的数组
	}
	输出
}