主要是利用分治的思想,根据中序遍历的结果找到分治的左子树和右子树序列,然后递归实现树的重建。
1.二叉树中序遍历与前序遍历构建二叉树
- 从前序与中序遍历序列构造二叉树
根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
可以假设树中没有重复的元素。
public TreeNode buildTree(int[] preorder, int[] inorder) {
if (inorder == null || preorder == null || inorder.length < 1 || preorder.length < 1) return null;
Map<Integer, Integer> inorderIndexMap = new HashMap<>();// 构建二叉树中序遍历数值和index的hash表,因为元素都不相同
for (int i = 0; i < inorder.length; i++) {
inorderIndexMap.put(inorder[i], i);
}
return helper(0, inorder.length - 1, 0, preorder.length - 1, inorder, preorder, inorderIndexMap);
}
public TreeNode helper(int inStart, int inEnd, int preStart, int preEnd, int[] inorder, int[] preOrder, Map<Integer, Integer> indexMap) {
if (inStart > inEnd || preStart > preEnd) return null;// 结束条件
TreeNode node = new TreeNode(preOrder[preStart]);
int mid = indexMap.get(preOrder[preStart]);
//容易知道 左子树中序遍历的开始和结束下标 [inStart, mid-1], 前序遍历的左边的元素个数和中序遍历左边的元素个数相同,容易得到前序遍历的start和end
node.left = helper(inStart, mid - 1, preStart + 1, preStart + 1 + (mid - 1 - inStart), inorder, preOrder, indexMap);
//容易知道 右子树中序遍历的开始和结束下标 mid+1, inEnd,
node.right = helper(mid + 1, inEnd, preEnd - (inEnd - mid - 1), preEnd, inorder, preOrder, indexMap);
return node;
}
2. 二叉树中序遍历与后序遍历构建二叉树
- 从中序与后序遍历序列构造二叉树
根据一棵树的中序遍历与后序遍历构造二叉树。
这里是引用
可以假设树中没有重复的元素。
/**
* 分治的思想
* @param inorder
* @param postOrder
* @return
*/
public TreeNode buildTree(int[] inorder, int[] postOrder) {
if (inorder == null || postOrder == null || inorder.length < 1 || postOrder.length < 1) return null;
Map<Integer, Integer> inorderIndexMap = new HashMap<>();// 构建二叉树中序遍历数值和index的hash表
for (int i = 0; i < inorder.length; i++) {
inorderIndexMap.put(inorder[i], i);
}
return helper(0, inorder.length - 1, 0, postOrder.length - 1, inorder, postOrder, inorderIndexMap);
}
public TreeNode helper(int inStart, int inEnd, int postStart, int postEnd, int[] inorder, int[] postOrder, Map<Integer, Integer> indexMap) {
if (inStart > inEnd || postStart > postEnd) return null;// 结束条件
TreeNode node = new TreeNode(postOrder[postEnd]);
int mid = indexMap.get(postOrder[postEnd]);
//容易知道 左子树中序遍历的开始和结束下标 [inStart, mid-1],
// 后续遍历的左边的元素个数和中序遍历左边的元素个数相同,所以后续遍历的左边的结束下标是mid-1 - inStart+postStart
node.left = helper(inStart, mid - 1, postStart, mid - 1 - inStart + postStart, inorder, postOrder, indexMap);
//容易知道 右子树中序遍历的开始和结束下标 mid+1, inEnd,
node.right = helper(mid + 1, inEnd, mid - 1 - inStart + postStart + 1, postEnd - 1, inorder, postOrder, indexMap);
return node;
}