给你一个正整数数组 arr,考虑所有满足以下条件的二叉树:
每个节点都有 0 个或是 2 个子节点。
数组 arr 中的值与树的中序遍历中每个叶节点的值一一对应。(知识回顾:如果一个节点有 0 个子节点,那么该节点为叶节点。)
每个非叶节点的值等于其左子树和右子树中叶节点的最大值的乘积。
在所有这样的二叉树中,返回每个非叶节点的值的最小可能总和。这个和的值是一个 32 位整数。
示例:
输入:arr = [6,2,4]
输出:32
解释:
有两种可能的树,第一种的非叶节点的总和为 36,第二种非叶节点的总和为 32。
提示:
2 <= arr.length <= 40
1 <= arr[i] <= 15
答案保证是一个 32 位带符号整数,即小于 2^31。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-cost-tree-from-leaf-values
题解
根据题意,较小的叶子节点要尽量放在底部,因为底部被乘的次数更多。单调递减栈可以找到一个极小值,当我们发现一个点的值比栈顶元素要大,说明在我们的策略下栈顶元素肯定要和左边的点或者右边的点(栈中下一个点和当前点)组成左子树和右子树,那么我们要选择更小的点与栈顶元素配对。
代码
class Solution {
public:
int mctFromLeafValues(vector<int>& arr) {
int res = 0;
stack<int> s;
s.push(INT_MAX);
int n = arr.size();
for (int i=0;i<n;i++) {
while (s.top() <= arr[i]) {
int temp = s.top();
s.pop();
res += temp * min(s.top(), arr[i]);
}
s.push(arr[i]);
}
while (s.size() > 2) {
int temp = s.top();
s.pop();
res += temp * s.top();
}
return res;
}
};