传送门
NOIP 2003 D1T3(区间/树形dp)
(l,mid-1),(mid+1,r)即mid为根的左右子树。
dp[l][r]=max{dp[l][mid-1]*dp[mid+1][r]+point[mid]}
dp时再记录最优解选择的根结点即可。
注意:root[i][i]=i。否则输出前序遍历要崩!
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int MAXN=32;
int n;
int dp[MAXN][MAXN],root[MAXN][MAXN];
void print(int l,int r) {
if (l>r) return ;
printf("%d ",root[l][r]);
print(l,root[l][r]-1);
print(root[l][r]+1,r);
}
int main() {
// freopen("P1040.in","r",stdin);
memset(dp,0,sizeof(dp));
scanf("%d",&n);
for (int i=1;i<=n;++i) scanf("%d",&dp[i][i]),root[i][i]=i,dp[i][i-1]=1;
dp[n+1][n]=1;
for (int i=n-1;i;--i)
for (int j=i+1;j<=n;++j)
for (int k=i;k<=j;++k)
if (dp[i][j]<dp[i][k-1]*dp[k+1][j]+dp[k][k])
dp[i][j]=dp[i][k-1]*dp[k+1][j]+dp[k][k],root[i][j]=k;
printf("%d\n",dp[1][n]);
print(1,n);
return 0;
}