洛谷 P1040 加分二叉树

link

思路整理

中序遍历有一个很重要的性质,就是一颗树的中序遍历 [L,R] 的一个连续的子区间,一定对应它的一颗子树,我们用 [L,K] 表示

自然,这颗二叉树的子树就是 [L,K-1],[K+1,R] (根节点 K )

然后,对于每个区间我们都要枚举 k ,以便求出最大得分

很容易想到,f[l][r]=max{f[l][k-1]*w[k]*f[k+1][r]}

于是,难点就在于怎样求出该方案 (答案不唯一,建议SPJ)

可以用 g[l][r]表示区间 [L,R] 对应的根节点,来记录每次的决策

这样每次先令 rt=g[l][r],输出rt,再递归输出区间 [L,rt-1] , [rt+1,R]

既然没有SPJ ,那只好输出字典序最小的方案了

由于是前序遍历,我们就要是根节点字典序尽量小(因为先输出)

体现在代码中,就是从L到R枚举根节点是,如果方案更优就更新,如果方案更劣或效果一样就不更新

然后…

AC代码

/*
数组没开够，爆零两行泪
ll开成 int，爆零两行泪
多组忘清空，爆零两行泪
 dp 没初值，爆零两行泪
深搜没边界，爆零两行泪
广搜忘出队，爆零两行泪
输入没加 &，爆零两行泪
模数没看见，爆零两行泪
 -1 不输出，爆零两行泪
越界不特判，爆零两行泪
空间开一倍，爆零两行泪
无向变有向，爆零两行泪
题意没审清，爆零两行泪
文件名起错，爆零两行泪
调试忘删除，爆零两行泪
文件不保存，爆零两行泪
文件不读入，爆零两行泪
文件不输出，爆零两行泪
*/
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=55;
int n;
ll f[N][N];
int w[N];
int g[N][N];
void dfs(int l,int r){
	if(l>r) return;
	int rt=g[l][r];
	printf("%d ",rt);
	dfs(l,rt-1);
	dfs(rt+1,r);
	return;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",w+i);
	for(int len=1;len<=n;len++)
		for(int l=1;l+len-1<=n;l++){
			int r=l+len-1;
			if(len==1){
				f[l][r]=w[l];
				g[l][r]=l;
				continue;//death point:f[l][r]会莫名其妙的多1(1*1+w[l]) 
			}
			for(int k=l;k<=r;k++){
				int left= k==l?1:f[l][k-1];
				int right= k==r?1:f[k+1][r];
				ll score=left*right+w[k];
				if(score>f[l][r]){
					f[l][r]=score;
					g[l][r]=k;
				}
			}
			
		}
	printf("%lld\n",f[1][n]);
	dfs(1,n);
	puts("");
	return 0;
}