题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
这道题就比较经典了,经典递归。
JS版:(我加了测试)
function TreeNode(x) {
this.val = x;
this.left = null;
this.right = null;
}
function reConstructBinaryTree(pre, vin)
{
/*不要想复杂,其实就是不断递归压缩前序数组和中序数组的过程,直到分到的左右子树为null为止,即这时候前序和
中序数组都是空的。
*/
// console.log(pre);
// console.log(vin);
if(pre.length==0||vin.length==0)
return null;
//前序为1 2 4 7 3 5 6 8
//后台为4 7 2 1 5 3 8 6
var root = pre[0];
var index = vin.indexOf(root);
var leftArr = vin.slice(0,index);
var rightArr = vin.slice(index + 1);
// 当只剩下单节点的时候,
//选择根,继续分开左子树和右子树,递归
var node = new TreeNode(root);
node.left = reConstructBinaryTree(pre.slice(1,index+1),leftArr);
node.right = reConstructBinaryTree(pre.slice(index+1),rightArr);
return node;
}
pre = [1,2,4,7,3,5,6,8];
vin = [4,7,2,1,5,3,8,6];
var Node = reConstructBinaryTree(pre,vin);
console.log(Node);
C++版:
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
return reConBTree(pre,0,pre.length-1,in,0,in.length-1);
}
public TreeNode reConBTree(int [] pre,int preleft,int preright,int [] in,int inleft,int inright){
if(preleft > preright || inleft> inright)//当到达边界条件时候返回null
return null;
TreeNode root = new TreeNode(pre[preleft]);
for(int i = inleft; i<= inright; i++){
if(pre[preleft] == in[i]){
root.left = reConBTree(pre,preleft+1,preleft+i-inleft,in,inleft,i-1);
root.right = reConBTree(pre,preleft+i+1-inleft,preright,in,i+1,inright);
}
}
return root;
}
}