根據二叉樹前序遍歷和中序遍歷還原二叉樹Java實現

---

title: 根據二叉樹前序遍歷和中序遍歷還原二叉樹Java實現
date: 2019-07-14
tags: java

根據二叉樹前序遍歷和中序遍歷還原二叉樹Java實現

問題描述：
已知二叉樹的前序遍歷爲{1,2,4,7,3,5,6,8}
中序遍歷爲{4,7,2,1,5,3,8,6}
還原二叉樹並返回（根節點）

---

思路：
已知二叉樹的前序遍歷和中序遍歷如何還原二叉樹？
從前序遍歷的概念來說：根節點-左節點-右節點
中序遍歷則爲：左節點-根節點-右結點
對於整個二叉樹來說，顯而易見前序遍歷的第一個元素值就是整個二叉樹的根節點

---

所以在前序遍歷{1,2,4,7,3,5,6,8}中，1爲整個二叉樹的根節點。
而對於中序遍歷中，根節點位置的左邊一定是所有左子樹節點，右邊一定是所有右子樹節點，所以在中序遍歷{4,7,2,1,5,3,8,6}中，已知1位根節點，所以{4,7,2}爲左子樹，{5,3,8,6}爲右子樹。

---

這樣，相對的又得到兩顆二叉樹（左子樹和右子樹）
會看整個前序遍歷，已知左子樹爲{4,7,2}，在前序遍歷中是{2,4,7}，所以說明在左子樹中，2位根節點，4,爲左節點，7爲右節點。

---

而對於右子樹{5,3,8,6}，在前序遍歷中的順序是{3,5,6,8}，說明3爲右子樹根節點，而對於本身就處於最初中序遍歷位置的右子樹{5,3,8,6}，說明相對的，3左邊爲右子樹的左節點，8,6則爲右子樹的右子樹

---

…同樣的劃分下去，就會有個遞歸的思想

---

代碼實現：

package com.lagoon.test;

import java.util.Arrays;

/**
 * @Author WinkiLee
 * @Date 2019/7/14 13:40
 * @Description
 */
public class Solution {

        public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
            if(pre.length == 0||in.length == 0){
                return null;
            }
            TreeNode node = new TreeNode(pre[0]);
            for(int i = 0; i < in.length; i++){
                if(pre[0] == in[i]){
                    node.left = reConstructBinaryTree(Arrays.copyOfRange(pre, 1, i+1), Arrays.copyOfRange(in, 0, i));
                    node.right = reConstructBinaryTree(Arrays.copyOfRange(pre, i+1, pre.length), Arrays.copyOfRange(in, i+1,in.length));
                }
            }
            return node;
        }

    public static void main(String[] args) {
        int[] a = {1, 2, 4, 7, 3, 5, 6, 8};
        int[] b = {4, 7, 2, 1, 5, 3, 8, 6};
        Solution solution = new Solution();
        TreeNode t=solution.reConstructBinaryTree(a, b);
        System.out.println(t.left.val);
    }
}

輸出結果應該取到的是根節點的左節點的值，應該是2
輸出截圖