Luogu 1040 加分二叉树（区间dp）

最新推荐文章于 2025-01-27 11:51:56 发布

原创最新推荐文章于 2025-01-27 11:51:56 发布 · 248 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#区间dp

收录于

区间dp 专栏收录该内容

2 篇文章

订阅专栏

本文深入探讨了区间动态规划算法在解决特定类型问题上的应用。通过一个具体的实例，讲解了如何利用区间DP算法来求解最优解，包括状态定义、状态转移方程、递推过程以及最终结果的输出。文章还提供了完整的代码实现，帮助读者更好地理解和掌握区间DP算法。

传送门

题解：区间dp，记录取值路径

f[i][j]表示从i到j为一棵树能获得的最大加分，rt[i][j]表示从i到j为一棵树能获得最大加分时的根节点编号

调了半天发现叶节点不用+1......π__π

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=34;
int n;
ll d[N],f[N][N];
int rt[N][N];
inline void smax(int i,int j,int k,int v) {
	if (f[i][j]<v) {
		f[i][j]=v;
		rt[i][j]=k;
	}
}
void print(int l,int r) {
	if (l>r) return ;
	printf("%d ",rt[l][r]);
	if (l==r) return ;
	print(l,rt[l][r]-1);
	print(rt[l][r]+1,r);
}
int main() {
	scanf("%d",&n);
	for (int i=1;i<=n;++i) {
		scanf("%lld",&d[i]);
		rt[i][i]=i;
		f[i][i]=d[i];
	}
	for (int len=2;len<=n;++len)
		for (int i=1;i+len-1<=n;++i) {
			int j=i+len-1;
			for (int k=i;k<=j;++k) {
				if (k==i) smax(i,j,k,d[k]+f[k+1][j]);
				else if (k==j) smax(i,j,k,d[k]+f[i][k-1]);
				else smax(i,j,k,d[k]+f[i][k-1]*f[k+1][j]);
			}
		}
	printf("%lld\n",f[1][n]);
	print(1,n);
	return 0;
}