【单调栈】力扣1130. 叶值的最小代价生成树

最新推荐文章于 2025-04-03 15:29:45 发布

hlc@

最新推荐文章于 2025-04-03 15:29:45 发布

阅读量231

点赞数 1

分类专栏：数据结构待完善文章标签： leetcode 算法职场和发展

本文链接：https://blog.youkuaiyun.com/sjsjs11/article/details/145195621

版权

数据结构同时被 2 个专栏收录

142 篇文章

订阅专栏

待完善

6 篇文章

订阅专栏

给你一个正整数数组 arr，考虑所有满足以下条件的二叉树：

每个节点都有 0 个或是 2 个子节点。
数组 arr 中的值与树的中序遍历中每个叶节点的值一一对应。
每个非叶节点的值等于其左子树和右子树中叶节点的最大值的乘积。
在所有这样的二叉树中，返回每个非叶节点的值的最小可能总和。这个和的值是一个 32 位整数。

如果一个节点有 0 个子节点，那么该节点为叶节点。

示例 1：
在这里插入图片描述
输入：arr = [6,2,4]
输出：32
解释：有两种可能的树，第一种的非叶节点的总和为 36 ，第二种非叶节点的总和为 32 。

示例 2：
在这里插入图片描述
输入：arr = [4,11]
输出：44

在这里插入图片描述

单调栈

class Solution {
public:
    int mctFromLeafValues(vector<int>& arr) {
        int res = 0;
        stack<int> stk;
        for (int x : arr) {
            while (!stk.empty() && stk.top() <= x) {
                int y = stk.top();
                stk.pop();
                if (stk.empty() || stk.top() > x) {
                    res += y * x;
                } else {
                    res += stk.top() * y;
                }
            }
            stk.push(x);
        }
        while (stk.size() >= 2) {
            int x = stk.top();
            stk.pop();
            res += stk.top() * x;
        }
        return res;
    }
};