LeetCode刷题笔记 1130. 叶值的最小代价生成树

题目描述

给你一个正整数数组 arr，考虑所有满足以下条件的二叉树：

1. 每个节点都有 0 个或是 2 个子节点。
2. 数组 arr 中的值与树的中序遍历中每个叶节点的值一一对应。（如果一个节点有 0 个子节点，那么该节点为叶节点。）
3. 每个非叶节点的值等于其左子树和右子树中叶节点的最大值的乘积。

在所有这样的二叉树中，返回每个非叶节点的值的最小可能总和。这个和的值是一个 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。