C++刷LeetCode（1130. 叶值的最小代价生成树）【递减栈】【数学】
题目描述：给你一个正整数数组 arr，考虑所有满足以下条件的二叉树：
每个节点都有 0 个或是 2 个子节点。
数组 arr 中的值与树的中序遍历中每个叶节点的值一一对应。（知识回顾：如果一个节点有 0 个子节点，那么该节点为叶节点。）
每个非叶节点的值等于其左子树和右子树中叶节点的最大值的乘积。
在所有这样的二叉树中，返回每个非叶节点的值的最小可能总和。这个和的值是一个 32 位整数。
示例：
输入：arr =
输出：32
解释：
有两种可能的树，第一种的非叶节点的总和为 36，第二种非叶节点的总和为 32。
24 24
/\ /\
12 4 6 8
/\ / \
6 2 2 4
提示：
2 <= arr.length <= 40
1 <= arr <= 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>store;
store.push(INT_MAX);
for(auto cha : arr){
while(cha > store.top()){
int temp = store.top();
store.pop();
res += temp * min(store.top(), cha);
}
store.push(cha);
}
while(store.size() > 2){
int temp = store.top();
store.pop();
res += temp * store.top();
}
return res;
}
};
参考链接：https://leetcode-cn.com/problems/minimum-cost-tree-from-leaf-values/solution/c-dan-diao-zhan-by-crossing-2/
页:
[1]