1. 程式人生 > >[leetcode] 106. 從中序與後序遍歷序列構造二叉樹

[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;
    }

}