剑指Offer-重建二叉树
题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树
3
/ \
9 20
/ \
15 7
问题分析
前序遍历:根节点->左子树->右子树
中序遍历:左子树->根节点->右子树
首先要找出三个根节点:根结点、左子树根节点、右子树根节点。
- 前序遍历的首元素为根结点node
- 中序遍历中找到根节点node的索引值,node将其分为左子树|根节点|右结点
- 根据中序遍历得到的左子树和右子树数量,可以将前序遍历分为根节点|左子树|右子树
对于树的左右子树,采用递归分治的思想找到左右子树根节点。
算法
首先,我们定义根节点在前序遍历的索引root,子树在中序遍历中的左边界left和有边界right
递归边界:left>right,终止递归,越过叶结点,返回null
- 建立根节点 node : 节点值为 preorder[root] ;
- 划分左右子树: 查找根节点在中序遍历 inorder 中的索引 i ;
- 构建左右子树:开启左右子树递归
- 回溯返回:回溯返回node,作为上一个根节点的左右子结点
问题解决
int[] preorder;
HashMap<Integer, Integer> dic = new HashMap<>();
public TreeNode buildTree(int[] preorder, int[] inorder) {
this.preorder = preorder;
for(int i = 0; i < inorder.length; i++)
dic.put(inorder[i], i);
return recur(0, 0, inorder.length - 1);
}
TreeNode recur(int root, int left, int right) {
if(left > right) return null; // 递归终止
TreeNode node = new TreeNode(preorder[root]); // 建立根节点
int i = dic.get(preorder[root]); // 划分根节点、左子树、右子树
node.left = recur(root + 1, left, i - 1); // 开启左子树递归
node.right = recur(root + i - left + 1, i + 1, right); // 开启右子树递归
return node; // 回溯返回根节点
}