刷怪日记-从中序与后序遍历序列构造二叉树

题目

解题思路

• 中序遍历为左根右，后序遍历为左右根
• 构建二叉树可以采用自顶向下递归，类似先序遍历递归形式，先确定根节点，然后分别确定该节点左右子树
• 大体思路：观察中序和后序遍历数组，可以发现后序遍历数组末尾元素往往是根节点，例如末尾元素3就是根节点，而元素3在中序遍历数组中刚好可以切割数组为左右子树俩部分。
• 具体递归思路如下
  1. 第一步：如果数组大小为零，说明为空节点，返回null
  2. 第二步：如果不为空，那么取后序数组末尾元素作为根节点元素
  3. 第三步：找到后序数组末尾元素在中序数组索引index，并作为切割点
  4. 第四步：按照index将中序数组切分为左右俩部分，在此算法中采用左闭右开原则，分别为[inBegin, index) [index, inEnd)
  5. 第五步：切割后序数组，切成后序左数组和后序右数组
  6. 第六步：递归处理左区间和右区间

 Map<Integer, Integer> map;
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        
        map = new HashMap<>(); // 使用map记录中序数组(节点值和索引)，便于递归查找切割点索引
    	// 记录Entry(节点值，索引)
        for(int i = 0; i < postorder.length; i++) map.put(inorder[i], i);

        return method(inorder, 0, postorder.length, postorder, 0, postorder.length);
    }


    public TreeNode method(int[] inorder, int inBegin, int inEnd, int[] postorder, int postBeign, int postEnd){

        
        if (inBegin >= inEnd || postBeign >= postEnd) {  // 不满足左闭右开，说明没有元素，返回空树
            return null;
        }
        
        int index = map.get(postorder[postEnd-1]); // 取后序遍历末尾元素
        int leftNum = index - inBegin; // 左中序遍历节点个数
        TreeNode root = new TreeNode(inorder[index]);
		// 切割中序数组 左闭右开原则
        //左子树
        int inLeftBegin = inBegin;
        int inLeftEnd = index;
        // 右子树
   		int inRightBegin = index+1;
        int inRightEnd = inEnd;
        // 切割后序数组 左闭右开原则
        // 左子树
        int postLeftBegin = postBegin;
        int postLeftEnd = postBegin+leftNum; //起始位置+中序数组左子树节点个数
        // 右子树
        int postRightBegin = postBegin+leftNum;
        int postRightEnd = postEnd - 1; // 去除末尾元素，因为末尾元素已经初始化为根节点
        
        root.left = method(inorder, inLeftBegin, inLeftEnd, postorder, postLeftBegin, postLeftEnd);
        root.right = method(inorder, inRightBegin, inRightEnd, postorder, postRightBegin, postRightEnd);
        
        return root;
    }

​ 上述代码简洁版

  Map<Integer, Integer> map;
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        
        map = new HashMap<>();
    
        for(int i = 0; i < postorder.length; i++) map.put(inorder[i], i);

        return method(inorder, 0, postorder.length, postorder, 0, postorder.length);
    }


    public TreeNode method(int[] inorder, int inBegin, int inEnd, int[] postorder, int postBeign, int postEnd){

        
        if (inBegin >= inEnd || postBeign >= postEnd) {  // 不满足左闭右开，说明没有元素，返回空树
            return null;
        }
        
        int index = map.get(postorder[postEnd-1]); // 取后序遍历末尾元素
        int leftNum = index - inBegin; // 左中序遍历节点个数
        TreeNode root = new TreeNode(inorder[index]);

        root.left = method(inorder, inBegin, index, postorder, postBeign, postBeign+leftNum);
        root.right = method(inorder, index+1, inEnd, postorder, postBeign+leftNum, postEnd-1);
        
        return root;
    }