剑指offer(四) 重建二叉树
来源:互联网 发布:资本流入本币升值知乎 编辑:程序博客网 时间:2024/05/22 14:37
题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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; }}
阅读全文