2019-11-02 12:07:41 +08:00
|
|
|
|
# 7. 重建二叉树
|
|
|
|
|
|
|
2019-11-02 21:22:13 +08:00
|
|
|
|
## 题目链接
|
|
|
|
|
|
|
|
|
|
|
|
[牛客网](https://www.nowcoder.com/practice/8a19cbe657394eeaac2f6ea9b0f6fcf6?tpId=13&tqId=11157&tPage=1&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking&from=cyc_github)
|
2019-11-02 12:07:41 +08:00
|
|
|
|
|
|
|
|
|
|
## 题目描述
|
|
|
|
|
|
|
|
|
|
|
|
根据二叉树的前序遍历和中序遍历的结果，重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
|
|
|
|
|
|
|
|
|
|
|
|
|
2019-11-02 21:22:13 +08:00
|
|
|
|
|
2019-12-06 01:04:29 +08:00
|
|
|
|
<div align="center"> <img src="https://cs-notes-1256109796.cos.ap-guangzhou.myqcloud.com/image-20191102210342488.png" width="400"/> </div><br>
|
2019-11-02 12:07:41 +08:00
|
|
|
|
|
|
|
|
|
|
## 解题思路
|
|
|
|
|
|
|
2019-11-02 21:22:13 +08:00
|
|
|
|
前序遍历的第一个值为根节点的值，使用这个值将中序遍历结果分成两部分，左部分为树的左子树中序遍历结果，右部分为树的右子树中序遍历的结果。然后分别对左右子树递归地求解。
|
2019-11-02 12:07:41 +08:00
|
|
|
|
|
2019-12-06 01:04:29 +08:00
|
|
|
|
<div align="center"> <img src="https://cs-notes-1256109796.cos.ap-guangzhou.myqcloud.com/60c4a44c-7829-4242-b3a1-26c3b513aaf0.gif" width="430px"> </div><br>
|
2019-11-02 12:07:41 +08:00
|
|
|
|
|
|
|
|
|
|
```java
|
|
|
|
|
|
// 缓存中序遍历数组每个值对应的索引
|
|
|
|
|
|
private Map<Integer, Integer> indexForInOrders = new HashMap<>();
|
|
|
|
|
|
|
|
|
|
|
|
public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
|
|
|
|
|
|
for (int i = 0; i < in.length; i++)
|
|
|
|
|
|
indexForInOrders.put(in[i], i);
|
|
|
|
|
|
return reConstructBinaryTree(pre, 0, pre.length - 1, 0);
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
private TreeNode reConstructBinaryTree(int[] pre, int preL, int preR, int inL) {
|
|
|
|
|
|
if (preL > preR)
|
|
|
|
|
|
return null;
|
|
|
|
|
|
TreeNode root = new TreeNode(pre[preL]);
|
|
|
|
|
|
int inIndex = indexForInOrders.get(root.val);
|
|
|
|
|
|
int leftTreeSize = inIndex - inL;
|
|
|
|
|
|
root.left = reConstructBinaryTree(pre, preL + 1, preL + leftTreeSize, inL);
|
|
|
|
|
|
root.right = reConstructBinaryTree(pre, preL + leftTreeSize + 1, preR, inL + leftTreeSize + 1);
|
|
|
|
|
|
return root;
|
|
|
|
|
|
}
|
|
|
|
|
|
```
|
2019-11-02 14:39:13 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2019-11-02 17:33:10 +08:00
|
|
|
|
<div align="center"><img width="320px" src="https://cs-notes-1256109796.cos.ap-guangzhou.myqcloud.com/githubio/公众号二维码-2.png"></img></div>
|
