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)
|
|||
|
|
|
|||
|
|
}
|