题目链接: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;
}