根据中序遍历与后序遍历还原二叉树

题目描述

根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。

用例描述：

中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树：
3
/
9 20
/
15 7

思路

1. 题目将中序遍历和后序遍历的结果以数组形式给出，不便于操作，因此可以用集合类来辅助完成二叉树的构建，这里我选用的是常用的list
2. 如果中序遍历的list长度为0，说明是空树直接返回nul
3. 如果中序遍历的list长度为1，说明当前节点是叶子节点，直接返回
4. 根据后序遍历的list，取到最后一个元素就是根节点
5. 根据中序遍历取到的第一个元素就是左子树位置记作leftsize
6. 根据中序遍历结果，可将其分为：
   leftIn [0,leftSize) ; rightIn [ leftSize+1,inOrderList.size() )
7. 根据后序遍历结果，可将其分为：
   leftPost [0,leftSize) ; rightPost [ leftSize,postOrderList.size()-1)；

代码如下:

import java.util.ArrayList;
import java.util.List;

class TreeNode{
    public int val;
    public TreeNode left;
    public TreeNode right;

    public TreeNode(int val) {
        this.val = val;
    }
}
class Solution {
    private List<Integer> arrayToList(int[] arr){
        List<Integer> list = new ArrayList<>();
        for (int x : arr){
            list.add(x);
        }
        return list;
    }
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        //1.将数组转化为list
        List<Integer> inOrderList = arrayToList(inorder);
        List<Integer> postOrderList = arrayToList(postorder);
        //2.构建辅助方法
        return buildHelper(inOrderList,postOrderList);
    }

    private TreeNode buildHelper(List<Integer> inOrderList, List<Integer> postOrderList) {
        //如果inOrderList.size() == 0,说明为空树，直接返回null
        if (inOrderList.size() == 0){
            return null;
        }
        //如果inOrderList.size() == 1,说明当前树只有一个节点，直接返回该节点即可
        if (inOrderList.size() == 1){
            return new TreeNode(inOrderList.get(0));
        }
        //postOrderLise.get(postOrderList.size()-1)表示根节点的值
        int rootVal = postOrderList.get(postOrderList.size()-1);
        TreeNode root = new TreeNode(rootVal);
        //根据rootVal在中序遍历中确定左子树的位置
        int leftSize = inOrderList.indexOf(rootVal);
        List<Integer> leftIn = inOrderList.subList(0,leftSize);
        List<Integer> rightIn = inOrderList.subList(leftSize+1,inOrderList.size());
        List<Integer> leftPost = postOrderList.subList(0,leftSize);
        List<Integer> rightPost = postOrderList.subList(leftSize,postOrderList.size()-1);
        //递归创建左子树
        root.left = buildHelper(leftIn,leftPost);
        //递归创建右子树
        root.right = buildHelper(rightIn,rightPost);
        return root;
    }
}