前序遍歷中序遍歷重建二叉樹

 

題目：

* 輸入某二叉樹的前序遍歷和中序遍歷的結果，請重建出該二叉樹。
* 假設輸入的前序遍歷和中序遍歷的結果中都不含重複的數字。
* 例如輸入前序遍歷序列{1,2,4,7,3,5,6,8}和中序遍歷序列{4,7,2,1,5,3,8,6}

思想：遞歸：f(n)的結果是根據f(n-1)的結果遞推過來的，

遞歸可以參考這篇文章：

https://baijiahao.baidu.com/s?id=1629571574350179349&wfr=spider&for=pc

本案例中：f(根節點) = f(左節點) +f(右節點)

因此就很簡單了，別忘記遞歸的結束條件：左右節點的前序和中序長度是1的時候。

public class 前序中序重建二叉樹 {

    public static void main(String[] args) {
        int [] pre = new int[]{1,2,3,4,5,6,7};
        int [] in = new int[]{3,2,4,1,6,5,7};
        TreeNode node = reConstructBinaryTree(pre,in);
    }

    public static TreeNode reConstructBinaryTree(int [] pre, int [] in) {

        if(pre.length <=0 || in.length <=0|| pre.length != in.length){
            return null;
        }
        int len = pre.length;
        if(len<=0){
            return  null;
        }

        int value = pre[0];
        TreeNode rootNode = createNewTreeNode(value,null,null);
        if(len ==1){
            return rootNode;
        }

        int valueLoc = getLocationInAarry(value,in);
        int[] leftPre = new int [valueLoc];
        int[] leftIn = new int [valueLoc];
        int[] rightPre = new int[len-1-valueLoc];
        int[] rightIn = new int[len-1-valueLoc];

        //左右子樹前序數組
        for(int i =1; i<len; ++i){
            if(i <= valueLoc){
                leftPre[i-1] = pre[i];
                continue;
            }
            rightPre[i-valueLoc-1] = pre[i];
        }

        //左右子樹中序數組
        for(int i =0; i<len; ++i){
            if(i<valueLoc){
                leftIn[i] = in[i];
                continue;
            }
            if(i ==valueLoc){
                continue;
            }
                rightIn[i-valueLoc-1] = in[i];
        }
        rootNode.left = reConstructBinaryTree(leftPre,leftIn);
        rootNode.right =reConstructBinaryTree(rightPre,rightIn);

        return rootNode;
    }

    static  int getLocationInAarry(int temp,int[] in){
        if(temp< 0 || in.length <=0){
            return 0;
        }
        for(int j =0;j< in.length; ++j){
            if(in[j] == temp){
                return j;
            }
        }
        return -1;
    }

    static TreeNode createNewTreeNode( int val,TreeNode left,  TreeNode right){
        TreeNode node = new TreeNode(val);
        node.left = left;
        node.right = right;
        return  node;
    }

}