P1040 加分二叉树

Posted 2020-12-13 foxc

P1040 加分二叉树

这是一个区间DP。
本题有一个一开始令我疑惑的点：为什么在第23行，这种解法只考虑了左子树为空的情况？后来想了想，觉得可能有两个原因。

1. 如果某一个节点之下有两个及以上的节点的话，左右子树不为空的情况一定比有一个子树为空的情况要优。第23行代码只是针对根节点下只有一个节点的情况。
2. 左子树为空和右子树为空这两种情况实际上是相同的（根节点的加分相同。）
   第一点我不知道如何证明。或许我的理解是错误的，还请路过的大佬不吝赐教。

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
ll a[40],d[40][40],n,root[40][40]; 
void print(ll l,ll r){
	if(l > r) return;
	cout << root[l][r] << " ";
	if(l == r) return;
	print(l,root[l][r] - 1);
	print(root[l][r] + 1,r);
}
int main(){
	cin >> n;
	for(int i = 1; i <= n; i++){
		cin >> a[i];
		d[i][i] = a[i];	
		root[i][i] = i;
	}
	ll maxn = 0;
	for(int l = 2; l <= n; l++)
		for(int i = 1; i + l - 1 <= n; i++){
			int j = i + l - 1;
			d[i][j] = d[i + 1][j] + d[i][i]; //假如左子树为空
			root[i][j] = i; 
			for(int k = i + 1 ; k < j; k++){ //枚举左右子树的中间断点
				if(d[i][j] < d[i][k - 1] * d[k + 1][j] + d[k][k]){
					d[i][j] = d[i][k - 1] * d[k + 1][j] + d[k][k];
					maxn = max(maxn,d[i][j]);
					root[i][j] = k; //记录根节点，用于输出方案
				}
			}
		} 
	cout << maxn << endl;
	print(1,n);
	return 0;
}