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