目录
显示
题目链接
https://www.luogu.org/problem/P1040
题解
一道树形 DP 的题,难度也不算太大。
首先考虑第一问:因为给的是中序遍历,所以可以枚举根节点,继而得出左右子树,然后按定义递归求解(这里要记忆化搜索,不然递归层数过多可能导致 TLE 或者是 RE)。
我们枚举根节点 \(i\),得到如下转移方程:
$$f_{l,r}=\max(f_{l,i-1} \times f_{i+1,r}+s_i)$$
然后是第二问:在更新最优解的时候,可以把该子树的根节点存下来,然后就可以递归输出了。
#include <stdio.h> #include <string.h> int f[305][305],ino[305],s[305],root[305][305]; int calc(int l,int r) { if(l>r)return 1;//空树 else if(~f[l][r])return f[l][r];//记忆化 else if(l==r)return s[l];//叶节点 int ans=0; for(int i=l;i<=r;i++)//枚举根节点 { int x=calc(l,i-1)*calc(i+1,r)+s[i]; if(x>ans) { root[l][r]=i; ans=x; } } return f[l][r]=ans; } void print_tree(int l,int r) { printf("%d ",root[l][r]); if(l<=root[l][r]-1)print_tree(l,root[l][r]-1);//输出左子树 if(root[l][r]+1<=r)print_tree(root[l][r]+1,r);//输出右子树 return; } int main() { int n; memset(f,-1,sizeof(f)); scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&s[i]); for(int i=1;i<=n;i++) root[i][i]=i; printf("%d\n",calc(1,n)); print_tree(1,n); return 0; }