71 lines
2.1 KiB
Java
71 lines
2.1 KiB
Java
//给你一个正整数数组 arr，考虑所有满足以下条件的二叉树：
|
||
//
|
||
//
|
||
// 每个节点都有 0 个或是 2 个子节点。
|
||
// 数组 arr 中的值与树的中序遍历中每个叶节点的值一一对应。（知识回顾：如果一个节点有 0 个子节点，那么该节点为叶节点。）
|
||
// 每个非叶节点的值等于其左子树和右子树中叶节点的最大值的乘积。
|
||
//
|
||
//
|
||
// 在所有这样的二叉树中，返回每个非叶节点的值的最小可能总和。这个和的值是一个 32 位整数。
|
||
//
|
||
//
|
||
//
|
||
// 示例：
|
||
//
|
||
// 输入：arr = [6,2,4]
|
||
//输出：32
|
||
//解释：
|
||
//有两种可能的树，第一种的非叶节点的总和为 36，第二种非叶节点的总和为 32。
|
||
//
|
||
// 24 24
|
||
// / \ / \
|
||
// 12 4 6 8
|
||
// / \ / \
|
||
//6 2 2 4
|
||
//
|
||
//
|
||
//
|
||
// 提示：
|
||
//
|
||
//
|
||
// 2 <= arr.length <= 40
|
||
// 1 <= arr[i] <= 15
|
||
// 答案保证是一个 32 位带符号整数，即小于 2^31。
|
||
//
|
||
// Related Topics 栈 树 动态规划
|
||
// 👍 163 👎 0
|
||
|
||
package leetcode.editor.cn;
|
||
|
||
import java.util.Arrays;
|
||
import java.util.Stack;
|
||
|
||
//1130:叶值的最小代价生成树
|
||
public class MinimumCostTreeFromLeafValues {
|
||
public static void main(String[] args) {
|
||
//测试代码
|
||
Solution solution = new MinimumCostTreeFromLeafValues().new Solution();
|
||
}
|
||
|
||
//力扣代码
|
||
//leetcode submit region begin(Prohibit modification and deletion)
|
||
class Solution {
|
||
public int mctFromLeafValues(int[] arr) {
|
||
Stack<Integer> stack = new Stack<>();
|
||
int result = 0;
|
||
stack.push(Integer.MAX_VALUE);
|
||
for (int num : arr) {
|
||
while (num > stack.peek()) {
|
||
result += stack.pop() * Math.min(stack.peek(), num);
|
||
}
|
||
stack.push(num);
|
||
}
|
||
while (stack.size() > 2) {
|
||
result += stack.pop() * stack.peek();
|
||
}
|
||
return result;
|
||
}
|
||
}
|
||
//leetcode submit region end(Prohibit modification and deletion)
|
||
|
||
} |