[leetcode] 106. 從中序與後序遍歷序列構造二叉樹
阿新 • • 發佈:2018-11-06
後序遍歷的第一個元素肯定是根節點,在中序遍歷中找到該元素,左半是該節點的左子樹的元素,右半是該節點的右子樹元素,記左半的長度為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; } }