首页 > 代码库 > Lintcode-Max Tree

Lintcode-Max Tree

Given an integer array with no duplicates. A max tree building on this array is defined as follow:

  • The root is the maximum number in the array
  • The left subtree and right subtree are the max trees of the subarray divided by the root number.
Construct the max tree by the given array.
Example

Given [2, 5, 6, 0, 3, 1], the max tree is

              6

            /    \

         5       3

       /        /   \

     2        0     1

 

 

Challenge

O(n) time complexity

Analysis:

Recursion: use recursion method, in the worst case, the complexity is O(n^2).

Linear method: Refer to the analysis of some other people: http://www.meetqun.com/thread-3335-1-1.html

这个题Leetcode上没有,其实这种树叫做笛卡树( Cartesian tree)。直接递归建树的话复杂度最差会退化到O(n^2)。经典建树方法,用到的是单调堆栈。我们堆栈里存放的树,只有左子树,没有有子树,且根节点最大。
(1) 如果新来一个数,比堆栈顶的树根的数小,则把这个数作为一个单独的节点压入堆栈。
(2) 否则,不断从堆栈里弹出树,新弹出的树以旧弹出的树为右子树,连接起来,直到目前堆栈顶的树根的数大于新来的数。然后,弹出的那些数,已经形成了一个新的树,这个树作为新节点的左子树,把这个新树压入堆栈。

这样的堆栈是单调的,越靠近堆栈顶的数越小
最后还要按照(2)的方法,把所有树弹出来,每个旧树作为新树的右子树。

Solution:

 

 1 /** 2  * Definition of TreeNode: 3  * public class TreeNode { 4  *     public int val; 5  *     public TreeNode left, right; 6  *     public TreeNode(int val) { 7  *         this.val = val; 8  *         this.left = this.right = null; 9  *     }10  * }11  */12 public class Solution {13     /**14      * @param A: Given an integer array with no duplicates.15      * @return: The root of max tree.16      */17     public TreeNode maxTree(int[] A) {18         if (A.length==0) return null;19 20         Stack<TreeNode> nodeStack = new Stack<TreeNode>();21         nodeStack.push(new TreeNode(A[0]));22         for (int i=1;i<A.length;i++)23             if (A[i]<=nodeStack.peek().val){24                 TreeNode node = new TreeNode(A[i]);25                 nodeStack.push(node);26             } else {27                 TreeNode n1 = nodeStack.pop();28                 while (!nodeStack.isEmpty() && nodeStack.peek().val < A[i]){29                     TreeNode n2 = nodeStack.pop();30                     n2.right = n1;31                     n1 = n2;32                 }33                 TreeNode node = new TreeNode(A[i]);34                 node.left = n1;35                 nodeStack.push(node);36             }37         38 39         TreeNode root = nodeStack.pop();40         while (!nodeStack.isEmpty()){41             nodeStack.peek().right = root;42             root = nodeStack.pop();43         }44 45         return root;46 47     48             49             50 51     }52 }

 

Solution 2 (recursion):

 1 /** 2  * Definition of TreeNode: 3  * public class TreeNode { 4  *     public int val; 5  *     public TreeNode left, right; 6  *     public TreeNode(int val) { 7  *         this.val = val; 8  *         this.left = this.right = null; 9  *     }10  * }11  */12 public class Solution {13     /**14      * @param A: Given an integer array with no duplicates.15      * @return: The root of max tree.16      */17     public TreeNode maxTree(int[] A) {18         if (A.length==0) return null;19 20         TreeNode root = new TreeNode(A[0]);21         for (int i=1;i<A.length;i++)22             if (A[i]>root.val){23                 TreeNode node = new TreeNode(A[i]);24                 node.left = root;25                 root = node;26             } else insertNode(root,null,A[i]);27 28         return root;29     }30 31     public void insertNode(TreeNode cur, TreeNode pre, int val){32         if (cur==null){33             TreeNode node = new TreeNode(val);34             pre.right = node;35             return;36         }37 38         if (cur.val<val){39             TreeNode node = new TreeNode(val);40             pre.right = node;41             node.left = cur;42             return;43         } else 44             insertNode(cur.right,cur,val);45     }46 47         48 }

 

 

Lintcode-Max Tree