VJP1100 加分二叉树（树形DP）

链接

归属树形DP  做着更像记忆化

DP很好做 就是那个输出路径恶心了。。改代码 从60多行改到120多行。。dp从1维加到三维。。

先类似记忆化搜索整棵树 枚举以i为根节点的最大值 子树类似

求完最大值 再递归搜一下前序 这题不记忆化会超时

 #include <iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<stdlib.h>
 using namespace std;
 #define LL long long
 LL dp[][][],a[];
 int n,path[],to;
 LL dfs(int root,int l,int r)
 {
     if(dp[root][l][r])
     return dp[root][l][r];
     if(l==r)
         return dp[root][l][r] = a[l];
     int i,j;
     LL tt =;
     if(l==root)
     {
         for(i = root+ ; i <= r ; i++)
             tt = max(tt,dfs(i,root+,r)+a[root]);
     }
     else if(r==root)
     {
         for(i = l ; i < root ; i++)
             tt = max(tt,dfs(i,l,root-)+a[root]);
     }
     else
     {
         for(i = l; i < root ; i++)
             for(j = root+ ; j <= r; j++)
             {
                 LL s1 = dfs(i,l,root-);
                 LL s2 = dfs(j,root+,r);
                 tt = max(tt,s1*s2+a[root]);
             }
     }
     return dp[root][l][r]=tt;
 }
 void find(int u,int l,int r,int v)
 {
     int i,j,x1,x2;
     if(l==r)
         return ;
     int f = ;
     if(u==l)
     {
         for(i = u+ ; i <= r ; i++)
         if(dp[i][u+][r]+a[u]==dp[u][l][r])
         {
             x1 = i;
             break;
         }
         path[v+] = x1;
         find(x1,u+,r,v+);
     }
     else if(u==r)
     {
         for(i = l ; i < u ; i++)
         if(dp[i][l][u-]+a[u]==dp[u][l][r])
         {
             x1 = i;
             break;
         }
         path[v+] = x1;
         find(x1,l,u-,v+);
     }
     else
     {
         for(i = l; i < u ; i++)
         {
             for(j = u+; j <= r ;j++)
             if(dp[i][l][u-]*dp[j][u+][r]+a[u]==dp[u][l][r])
             {
                 x1 = i;
                 x2 = j;
                 f = ;
                 break;
             }
             if(f) break;
         }
         path[v+] = x1;
         path[v++u-l] = x2;
         find(x1,l,u-,v+);
         find(x2,u+,r,v++u-l);
     }
 }
 int main()
 {
     int i;
     cin>>n;
     for(i = ; i <= n ; i++)
     cin>>a[i];
     LL ans = ;
     for(i = ; i <= n ; i++)
     {
         LL s = dfs(i,,n);
         if(s>ans)
         {
             ans =s;
             path[] = i;
         }
     }
     cout<<ans<<endl;
     find(path[],,n,);
     for(i = ; i < n ; i++)
     cout<<path[i]<<" ";
     cout<<path[n]<<endl;
     return ;
 }