[Swift]LeetCode1130. 叶值的最小代价生成树 | Minimum Cost Tree From Leaf Values

★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★
➤微信公众号：山青咏芝（shanqingyongzhi）
➤博客园地址：山青咏芝（https://www.cnblogs.com/strengthen/）
➤GitHub地址：https://github.com/strengthen/LeetCode
➤原文地址：http://www.javashuo.com/article/p-etpqnjeu-kv.html 
➤若是连接不是山青咏芝的博客园地址，则多是爬取做者的文章。
➤原文已修改更新！强烈建议点击原文地址阅读！支持做者！支持原创！
★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★

Given an array arr of positive integers, consider all binary trees such that:

• Each node has either 0 or 2 children;
• The values of arr correspond to the values of each leaf in an in-order traversal of the tree.  (Recall that a node is a leaf if and only if it has 0 children.)
• The value of each non-leaf node is equal to the product of the largest leaf value in its left and right subtree respectively.

Among all possible binary trees considered, return the smallest possible sum of the values of each non-leaf node.  It is guaranteed this sum fits into a 32-bit integer.

Example 1:

Input: arr = [6,2,4]
Output: 32
Explanation:
There are two possible trees.  The first has non-leaf node sum 36, and the second has non-leaf node sum 32.

    24            24
   /  \          /  \
  12   4        6    8
 /  \               / \
6    2             2   4

Constraints:

• 2 <= arr.length <= 40
• 1 <= arr[i] <= 15
• It is guaranteed that the answer fits into a 32-bit signed integer (ie. it is less than 2^31).

---

给你一个正整数数组 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。

---

12ms

 1 class Solution {
 2     func mctFromLeafValues(_ arr1: [Int]) -> Int {
 3         guard arr1.count > 0 else { return 0 }
 4         guard arr1.count > 1 else { return arr1[0] }
 5         
 6         var arr = [Int.max]
 7         arr.append(contentsOf: arr1)
 8         arr.append(Int.max)
 9         var sum = 0
10         while arr.count > 4 {
11             var mn = arr[1]
12             var idx = 1
13             for i in stride(from: 2, to: arr.count-1, by: 1) {
14                 if mn > arr[i] {
15                     mn = arr[i]
16                     idx = i
17                 }
18             }
19             var toMultiply = arr[idx-1] > arr[idx+1] ? arr[idx+1] : arr[idx-1]
20             sum += mn * toMultiply
21             arr.remove(at: idx)
22         }
23         
24         return sum + arr[1] * arr[2]
25     }
26 }

---

20ms

 1 class Solution {
 2     func mctFromLeafValues(_ arr: [Int]) -> Int {
 3         var solution = 0
 4         var currentArray = arr
 5         
 6         while currentArray.count > 1 {
 7             let nextIndex = currentArray.firstIndex(of: currentArray.min()!)!
 8             
 9             if nextIndex > 0 && nextIndex < currentArray.count - 1 {
10                 solution += currentArray[nextIndex] * 
11                             min(currentArray[nextIndex-1], currentArray[nextIndex+1])
12             } else {
13                 solution += currentArray[nextIndex] * 
14                             (nextIndex == 0 ? currentArray[nextIndex+1] : currentArray[nextIndex-1])
15             }
16             
17             currentArray.remove(at: nextIndex)
18         }
19         
20         
21         return solution
22     }
23 }

---

Runtime: 36 ms

Memory Usage: 20.9 MB

 1 class Solution {
 2     func mctFromLeafValues(_ arr: [Int]) -> Int {
 3         var dp:[[Int]] = [[Int]](repeating:[Int](repeating:0,count:45),count:45)
 4         var maxn:[[Int]] = [[Int]](repeating:[Int](repeating:0,count:45),count:45)
 5         var n:Int = arr.count
 6         for i in 0..<n
 7         {
 8             maxn[i][i] = arr[i]
 9             for j in (i + 1)..<n
10             {
11                  maxn[i][j] = max(maxn[i][j - 1], arr[j])
12             }
13         }
14         for d in 2...n
15         {
16             var i:Int = 0
17             while(i + d - 1 < n)
18             {
19                 var j:Int = i + d - 1
20                 dp[i][j] = (1<<60)
21                 for k in i..<j
22                 {
23                     dp[i][j] = min(dp[i][j],maxn[i][k] * maxn[k+1][j] + dp[i][k] + dp[k + 1][j])
24                 }
25                 i += 1
26             }
27         }
28         return (dp[0][n - 1])
29     }
30 }