树形DP or 单调栈贪心:叶值的最小代价生成树

树形DP or 单调栈贪心:叶值的最小代价生成树

问题:

在这里插入图片描述
在这里插入图片描述

思路:

树形DP:

对于题中所给数组,由于数组中的值与中序遍历中叶节点的值对应,则可将数组分为两部分,左部分为左子树叶节点,右部分为右子树叶节点,则根节点的值即为左右两部分数组中最大值的乘积,树中非叶节点和即为根节点的值+左子树非叶节点和+右子树非叶节点和,由于数组的划分是不确定的,因此需找到使得非叶节点和最小的划分

设DP[i][j]表示在原数组中左右界分别为i,j的数组得到的树的最小非叶节点和
M[i][j]表示在原数组中左右界分别为i,j的数组中的最大元素值
在这里插入图片描述
M初值为0,DP[i][i]=0表示树中仅有根节点,非叶节点和为0,其余DP的初值为+INF

DP的递推顺序为从最后一个元素向前递推,因为最后需求得的元素是DP[0][n-1],是从后往前推得到的

单调栈贪心:

这道题可以看做如下过程:在数组 arr 中,每次取相邻的两个数 a 和 b,然后去掉其中较小的一个,花费代价为 a * b,求最终将数组消减为一个元素的最小代价。那么,要想获得最小代价,我们应该采取的策略是:对于数组中的某一个数 a,分别向左和向右查询比它大的第一个数(为什么不是选择比它大的数中最小的数,因为需要满足中序遍历的顺序),在这两个数中选择较小的那个数进行组合并消去a,花费的代价最小。这个过程我们可以用单调栈来一次遍历解决掉。

维持一个单调栈,遍历数组元素,栈非空时,当遇到比栈顶元素大的元素,栈顶元素出栈,比较新的栈顶与当前遍历到的元素值,选择较小的一个组合,不断进行此操作直至栈顶元素值大于当前遍历到的元素值或栈为空。最后将当前元素入栈。

最后栈中元素个数若>=2(栈中元素为空或等于1表明已消减完成),说明在遍历的最后出现了连续递减的序列,此时不断将栈顶元素出栈并与新栈顶组合即可

代码:

树形DP:
class Solution {
public:
    int mctFromLeafValues(vector<int>& arr) {
        int n=arr.size();
        int DP[n][n];
        int M[n][n];
        memset( DP,1e+04,sizeof(DP) );
        memset( M,0,sizeof(M) );
        int i;
        int j;
        int k;
        int maxx;
        for( i=0;i<n;i++ )
        {
            maxx=0;
            for( j=i;j<n;j++ )
            {
                maxx=max( arr[j],maxx );
                M[i][j]=maxx;
            }
        }
        for( i=n-1;i>=0;i-- )
            for( j=i;j<n;j++ )
            {
                if( j==i )
                {
                    DP[i][j]=0;
                    continue;
                }
                for( k=i;k<=j-1;k++ )
                {
                   DP[i][j]=min( DP[i][j],M[i][k]*M[k+1][j]+DP[i][k]+DP[k+1][j] );
                }
            }
        return DP[0][n-1];
    }
};
单调栈贪心:
class Solution {
public:
    int mctFromLeafValues(vector<int>& arr) {
        stack<int> s;
        int ans=0;
        for( auto x:arr )
        {
            while( !s.empty() && s.top()<x )
            {
                int temp=s.top();
                s.pop();
                ans+=temp*min( s.empty()? 100:s.top(),x );
            }
            s.push( x );
        }
        while( s.size()>=2 )
        {
            int temp=s.top();
            s.pop();
            ans+=temp*s.top();
        }
        return ans;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

happy19991001

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值