剑指 Offer 07. 重建二叉树

重建二叉树

思路

在前序遍历中找到根节点的值，然后在中序遍历中根据根节点的值划分左右子树；
然后在左右子树里面递归调用同样的代码，再进行划分。。。

package swordPointingToTheOffer;

import java.util.HashMap;
import java.util.Map;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

public class seven {

    //映射节点值和下标
    Map<Integer,Integer> map = new HashMap<>();

    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if(preorder==null||preorder.length<=0){
            return null;
        }
        for(int i = 0;i<inorder.length;i++){
            map.put(inorder[i],i);
        }

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

        return root;
    }

    /**
     *
     * @param preorder 前序遍历的数组
     * @param l1        前序左下标
     * @param r1        前序右下标
     * @param inorder   中序遍历的数组
     * @param l2        中序左下标
     * @param r2        中序右下标
     * @return
     */
    TreeNode f(int[] preorder, int l1, int r1, int[] inorder ,int l2, int r2){
        //递归退出条件(左右子树都不存在的情况)
        if(l1>r1&&l2>r2){
            return null;
        }

        //获取根节点
        TreeNode root = new TreeNode(preorder[l1]);
        //获取根节点的下标 (中序遍历的下标)
        int i = map.get(preorder[l1]);

        //递归处理左子树
        root.left = f(preorder,l1+1,l1+(i-l2),inorder,l2,i-1);
        //递归处理右子树
        root.right = f(preorder,l1+(i-l2)+1,r1,inorder,i+1,r2);

        return root;

    }

    public static void main(String[] args) {

        int[] preorder = {3,9,5,6,20,15,7};
        int[]inorder = {5,9,6,3,15,20,7};
        seven seven = new seven();
        TreeNode root = seven.buildTree(preorder,inorder);
    }
}