LeetCode 105. Construct Binary Tree from Preorder and Inorder Traversal （用先序和中序树遍历来建立二叉树）

时间：2022-09-11 22:29:44

Given preorder and inorder traversal of a tree, construct the binary tree.

Note:
You may assume that duplicates do not exist in the tree.

题目标签：Array, Tree

　　题目给了我们preOrder 和 inOrder 两个遍历array，让我们建立二叉树。先来举一个例子，让我们看一下preOrder 和 inOrder的特性。

　　　　　　　　　　　　　　　　　　　　　　　　　　

　　　　　　　　　　　　　　　　　　　　　　　　　/　　\

　　　　　　　　　　　　　　　　　　　　　　　　

　　　　　　　　　　　　　　　　　　　　　　   /      \       /     \

　　　　　　　　　　　　

　　preOrder: 2 3 4 5 6 7   遍历顺序为：print， left， right

inOrder: 3 2 4 6 5 7     遍历顺序为： left， print， right

　　可以发现，preOrder的第一个肯定为root， 接下来都是左边的children， 在后面都是右边的children；

　　　　　　　inOrder的root在最中间，root左边的都是left children， root 右边的都是 right children。

　　所以，我们可以建立一个递归helper function来帮助建立二叉树，把root 传递给 helper function， 让helper function 递归每一个root， 并且建立每一个root的left 和right child；

　　其中的规律就是遍历preOrder array，

　　从 preOrder 里拿到root点，对于每一个点： 有一个范围，来判断是不是走到最底端走完了，如果走完了就return，没走完就继续；

　　　　　　　　　　　　　　　　　　　　　  然后利用inOrder里的位置关系，来找到这个root点的 left child 和right child；

　　　　　　　　　　　　　　　　　　　　　  还要缩小这个范围，如果是left child， 那么就在inOrder里缩小到左边去，如果是right child，那么就在inOrder里缩小到右边去。

　　举例：拿到1的话，设1为一个点，初始范围为inOrder position [0, 6] 接着要找到它的left child 和 right child;

　　　　　1的left -   preOrder中1的下一个数字2， 范围缩小到inOrder position [0, 2]，2在其中，设2为1的left， 递归2；

　　　　　2的left -   preOrder中2的下一个数字3， 范围缩小到inOrder position [0, 0]，3在其中，设3为2的left， 递归3；

　　　　　3的left -   preOrder中3的下一个数字4， 范围缩小到inOrder position [0, -1]，意味着走到底端了，返回到2；（3的right同理）

　　　　　2的right - preOrder中3的下一个数字4， 范围缩小到inOrder position [2, 2]，4在其中，设4为2的right，递归4；

　　　　　4的left -   preOrder中4的下一个数字5， 范围缩小到inOrder position [2, 1]，意味着走到底端了，返回到2 （4的right同理）；

　　　　　2的left right 都有了，所以继续返回到1；

　　　　　1的right - preOrder中4的下一个数字5， 范围缩小到InOrder position [4, 6]，5在其中，设5为1的right， 递归5；

　　　　　5的left -   preOrder中5的下一个数字6， 范围缩小到inOrder position [4, 4]，6在其中，设6为5的left， 递归6；

　　　　　6 走到底端，返回到5；

　　　　　5的right - preOrder中6的下一个数字7， 范围缩小到inOrder position [6, 6]，7在其中，设7为5的right， 递归7；

　　　　　7 走到低端，返回5；

　　　　　5 left right 都有，返回1；

　　　　　1 返回自己，结束。

　　具体细节请看code。

Java Solution:

Runtime beats 82.84%

完成日期：08/26/2017

关键词：Array, Tree

关键点：递归；利用pre-order 和 in-order 的位置关系递归

 /**

  * Definition for a binary tree node.

  * public class TreeNode {

  *     int val;

  *     TreeNode left;

  *     TreeNode right;

  *     TreeNode(int x) { val = x; }

  * }

  */

 class Solution

 {

     public TreeNode buildTree(int[] preorder, int[] inorder)

     {

         Map<Integer, Integer> inMap = new HashMap<Integer, Integer>();

         // save inorder number as key, position as value into map

         for(int i=0; i<inorder.length; i++)

             inMap.put(inorder[i], i);

         TreeNode root = helper(preorder, 0, 0, inorder.length - 1, inMap);

         return root;

     }

     public TreeNode helper(int[] preorder, int preStart, int inStart, int inEnd,

             Map<Integer, Integer> inMap)

     {

         if(inStart > inEnd)

             return null;

         int rootVal = preorder[preStart];

         TreeNode root = new TreeNode(rootVal);

         int inRoot = inMap.get(rootVal);  // position in inOrder

         /* inStart & inEnd: for left child, move inEnd to the left of root

          *                  for right child, move inStart to the right of root */

         root.left = helper(preorder, preStart + 1, inStart, inRoot - 1, inMap);

         /* preStart: for right child, go to inorder to check how many left children does root have,

          * add it into preorder to skip them to reach right child */

         root.right = helper(preorder, preStart + (inRoot - inStart) + 1, inRoot + 1, inEnd, inMap);

         return root;

     }

 }

参考资料：

https://discuss.leetcode.com/topic/3695/my-accepted-java-solution

LeetCode 算法题目列表 - LeetCode Algorithms Questions List

　　　　　　　　　　　　　

相关文章