[leetcode] 106. 从中序与后序遍历序列构造二叉树

106. 从中序与后序遍历序列构造二叉树

后序遍历的第一个元素肯定是根节点,在中序遍历中找到该元素,左半是该节点的左子树的元素,右半是该节点的右子树元素,记左半的长度为L,后序遍历中,前L个元素为左子树的元素,除去最后一个元素外的后半部分则为右子树的元素。递归处理即可。

class Solution {


    public TreeNode buildTree(int[] inorder, int[] postorder) {
        if (inorder == null || postorder == null
                || inorder.length == 0 || postorder.length == 0) {
            return null;
        }

        List<Integer> a = new ArrayList<>();
        List<Integer> b = new ArrayList<>();

        for (int i : inorder) {
            a.add(i);
        }

        for (int i : postorder) {
            b.add(i);
        }

        return buildTree(a, b);
    }

    public TreeNode buildTree(List<Integer> inorder, List<Integer> postorder) {
        TreeNode treeNode = null;

        Integer val = postorder.get(postorder.size() - 1);
        treeNode = new TreeNode(val);

        int i = inorder.indexOf(val);

        if (i > 0) {
            treeNode.left = buildTree(inorder.subList(0, i), postorder.subList(0, i));
        }
        if (inorder.size() - i - 1 > 0) {
            treeNode.right = buildTree(inorder.subList(i + 1, inorder.size()), postorder.subList(i, postorder.size() - 1));
        }

        return treeNode;
    }

}
posted @ 2018-11-06 20:14 ACBingo 阅读(...) 评论(...) 编辑 收藏