洛谷 P1040 加分二叉树

题目链接：https://www.luogu.org/problemnew/show/P1040
思路：寻找最大的权值，可以是DP，也可以用DFS来进行搜索，DFS（L， R）表示从L到R的最大权值的子树。那么就可以写出递归 :
temp = DFS（L， i - 1） * DFS （i + 1， R) + a[i]
这个递归就可以计算出的最大权值。用 num[L][R] 保存下来，同时再用另外一个数组记录计算最大权值所经过的结点。
代码 ：

#include <iostream>

using namespace std;

#define MAX_N 40

int n;
int a[MAX_N];
int num[MAX_N][MAX_N] = {1};
int ans[MAX_N][MAX_N] = {0}; // 用来储存结点 

int DFS (int l, int r) { // 表示从 l 到 r 的最大子树权值 
	if (l > r) return 1;
	else if (l == r){
		ans[l][r] = l; return a[l];
	}
	if (num[l][r]) return num[l][r];
	for (int i = l; i <= r; i++) {
		int temp = DFS (l , i - 1) * DFS (i + 1, r) + a[i];
		if (temp > num[l][r]) {
			ans[l][r] = i;
			num[l][r] = temp;
		}
	}
	return num[l][r];
}

void solve (int l, int r) {
	if (ans[l][r]) {
		cout << ans[l][r] << " ";
		solve (l, ans[l][r] - 1);
		solve (ans[l][r] + 1, r);	
	}
}

int  main() {
	cin >> n;
	for (int i = 1; i <= n; i++) cin >> a[i];
	DFS (1, n);
	cout << num[1][n] << endl;
	solve (1, n);
	return 0;
}