题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列 {1,2,4,7,3,5,6,8} 和中序遍历序列 {4,7,2,1,5,3,8,6},则重建二叉树并返回
题目链接:https://www.nowcoder.com/practice/8a19cbe657394eeaac2f6ea9b0f6fcf6
解题思路
根据前序遍历的特点,我们可以通过第一个值得到当前子树的根节点,借此去中序遍历查找根节点的左子树和右子树,采用递归的思路
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
// 获取根节点
int val = pre[0];
TreeNode root = new TreeNode(val);
// 得到左子树的大小
int len = 0;
for (; len < in.length; len++) {
if (in[len] == val) break;
}
// 构建左子树
if (len != 0) {
// 左子树的先序遍历
int[] pre_left = new int[len];
for (int i = 0; i < len; i++) {
pre_left[i] = pre[i+1];
}
// 左子树的中序遍历
int[] in_left = new int[len];
for (int i = 0; i < len; i++) {
in_left[i] = in[i];
}
root.left = reConstructBinaryTree(pre_left, in_left);
}
// 构建右子树
if (len != (pre.length-1)) {
// 右子树的先序遍历
int[] pre_right = new int[pre.length-len-1];
for (int i = 0, idx = len+1; idx < pre.length; i++, idx++) {
pre_right[i] = pre[idx];
}
// 右子树的中序遍历
int[] in_right = new int[in.length-len-1];
for (int i = 0, idx = len+1; idx < in.length; i++, idx++) {
in_right[i] = in[idx];
}
root.right = reConstructBinaryTree(pre_right, in_right);
}
return root;
}
}