题目描述
给你一个正整数数组 arr,考虑所有满足以下条件的二叉树:
- 每个节点都有 0 个或是 2 个子节点。
- 数组 arr 中的值与树的中序遍历中每个叶节点的值一一对应。(如果一个节点有 0 个子节点,那么该节点为叶节点。)
- 每个非叶节点的值等于其左子树和右子树中叶节点的最大值的乘积。
在所有这样的二叉树中,返回每个非叶节点的值的最小可能总和。这个和的值是一个 32 位整数。
示例:
输入:arr = [6,2,4]
输出:32
解释:
有两种可能的树,第一种的非叶节点的总和为 36,第二种非叶节点的总和为 32。
动态规划
class Solution {
public int mctFromLeafValues(int[] arr) {
int N = arr.length;
int[][] dp = new int[N][N];
int[][] maxVal = new int[N][N];
for(int i = 0; i < N; i++) {
for(int j = i; j < N; j++) {
int tmp = 0;
for(int k = i; k <= j; k++) {
tmp = Math.max(tmp, arr[k]);
}
maxVal[i][j] = tmp;
}
}
for(int[] a : dp)
Arrays.fill(a, Integer.MAX_VALUE);
for(int i = 0; i < N; i++) dp[i][i] = 0;
for(int len = 1; len < N; len++) {
for(int i = 0; i < N-len; i++) {
for(int j = i; j < i+len; j++) {
dp[i][i+len] = Math.min(dp[i][i+len],
dp[i][j] + dp[j+1][i+len] + (maxVal[i][j]*maxVal[j+1][i+len]
);
}
}
}
return dp[0] return dp[0][len-1];
}
}
自顶向下动态规划
class Solution {
int res;
int[] list;
public int mctFromLeafValues(int[] arr) {
int N = arr.length;
list = arr;
helper(0, N-1);
return res;
}
private void helper(int le, int ri) {
if(le >= ri) return;
int pos = maxPos(le, ri);
int max = getVal(pos);
helper(le, pos-1);
res += max * getVal(maxPos(le, pos-1));
helper(pos+1, ri);
res += max * getVal(maxPos(pos+1, ri));
}
private int maxPos(int le, int ri) {
if (le > ri) return -1;
int pos = le;
int max = list[pos];
for (int i = le; i <= ri; i++) {
if (list[i] > max) {
pos = i;
max = list[pos];
}
}
return pos;
}
private int getVal(int pos) {
return pos >= 0 ? list[pos] : 0;
}
}
单调栈(递减栈)
让大数尽可能的"少去乘"
class Solution {
public int mctFromLeafValues(int[] arr) {
Stack<Integer> stack = new Stack<>();
stack.push(Integer.MAX_VALUE);
int res = 0;
for(int n : arr) {
while(stack.peek() < n) res += stack.pop() * Math.min(stack.peek(), n);
stack.push(n);
}
while(stack.size() > 2) res += stack.pop() * stack.peek();
return res;
}
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-cost-tree-from-leaf-values
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。