loj 10151. 「一本通 5.1 练习 2」分离与合体

#10151. 「一本通 5.1 练习 2」分离与合体

题目链接

一道比较裸的区间dp

设状态 $f_{l,r}$ 为合并到 $[l,r]$ 时获得的最大价值，那么对于 $l\le k\le r$ 有

$$f_{l,r}=\max (f_{l,r},f_{l,k}+f_{k+1,r}+(a_l+a_r)\ast a_k)$$

枚举时按照区间dp的顺序，先枚举区间长度，再枚举左右端点，接着枚举区间范围内的数进行合并转移

题目还要求出分离顺序。我们可以在转移时，如果枚举到的区间合并方案优于先前的方案（即可以更新区间答案），那么就记录下 $p_{l,r}=k$，表示 $[l,r]$ 区间从 $k$ 位置分离。再开一个vector数组，每一个vector存储下标所对应层数的分离位置。然后递归扫描每一层的区间，将分离位置push_back进对应的vector中。输出时把每一层的分离位置sort后，从小到大输出即可。

Code

#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn=305;
int n,a[maxn],f[maxn][maxn],p[maxn][maxn],dep;
vector<int> ans[maxn];
void makePrint(int l,int r,int dep){
	if(l==r) return;                     //递归终止条件
	::dep=max(dep,::dep);            //记录下最大层数
	ans[dep].push_back(p[l][r]);
	makePrint(l,p[l][r],dep+1);      //递归搜索下一层
	makePrint(p[l][r]+1,r,dep+1);
}

int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int l=0;l<=n;l++){
		for(int i=1,j=i+l;j<=n;i++,j++){
			for(int k=i;k<j;k++){
				if(f[i][k]+f[k+1][j]+(a[i]+a[j])*a[k]>f[i][j]){
					f[i][j]=f[i][k]+f[k+1][j]+(a[i]+a[j])*a[k];
					p[i][j]=k;    //记录区间的分离位置
				}
			}
		}
	}
	printf("%d\n",f[1][n]);
	makePrint(1,n,1);          //递归将每一层的分离位置加入vector
	for(int i=1;i<=dep;i++) sort(ans[i].begin(),ans[i].end());      //对每一层的输出顺序排序
	for(int i=1;i<=dep;i++){   //按照层数优先的顺序输出
		int sz=ans[i].size();
		for(int j=0;j<sz;j++) printf("%d ",ans[i][j]);
	}
	putchar('\n');
	return 0;
}