标签:preorder 结点 遍历 TreeNode int 二叉树 null 重建
题目描述
输入一棵二叉树前序遍历和中序遍历的结果,请重建该二叉树。
样例
给定:
前序遍历是:[3, 9, 20, 15, 7]
中序遍历是:[9, 3, 15, 20, 7]
返回:[3, 9, 20, null, null, 15, 7, null, null, null, null]
返回的二叉树如下所示:
3
/ \
9 20
/ \
15 7
解法
在二叉树的前序遍历序列中,第一个数字总是根结点的值。
在中序遍历序列中,根结点的值在序列的中间,左子树的结点位于根结点左侧,而右子树的结点位于根结点值的右侧。
遍历中序序列,找到根结点,递归构建左子树与右子树。
由上图可知,前序遍历序列的第一个数字3就是根结点的值。扫描中序遍历序列,就能确定根结点的位置。
根据中序遍历的特点,在根结点的值3前面有1个数字是左子树结点的值,位于3后面的数字都是右子树结点的值。
同理对每个子树作同样的操作,使用递归
package demo; import java.util.ArrayList; import java.util.PriorityQueue; import java.util.Stack; import java.util.Comparator; public class Solution { class TreeNode{ int val; TreeNode left; TreeNode right; TreeNode(int x){ val = x; } } public TreeNode buildTree(int[] preorder, int[] inorder){ if (preorder == null || inorder == null || preorder.length == 0 || preorder.length != inorder.length) { return null; } return build(preorder, 0, preorder.length-1, inorder, 0, inorder.length-1); } private TreeNode build(int[] preorder, int startPre, int endPre, int[] inorder, int startIn, int endIn ){ if(startPre>endPre || startIn>endIn) return null; TreeNode root = new TreeNode(preorder[startPre]); //树根结点 for(int i=startIn; i<=endIn; i++){ if(inorder[i] == preorder[startPre]){ //注意边界:i-startIn为偏移量 root.left = build(preorder, startPre+1, startPre+i-startIn, inorder, startIn, i-1); root.right = build(preorder, i-startIn+startPre+1, endPre, inorder, i+1, endIn); } } return root; } public static void printPreOrder(TreeNode head) { if (head != null) { System.out.println(head.val); printPreOrder(head.left); printPreOrder(head.right); } } public static void main(String[] args) { int[] preorder = {1,2,4,7,3,5,6,8}; int[] inorder = {4,7,2,1,5,3,8,6}; Solution rt = new Solution(); Solution.TreeNode root = rt.buildTree(preorder, inorder); rt.printPreOrder(root); } }
https://github.com/kexun/jianzhioffer/blob/master/src/com/offer/Day03.java
待完善...
标签:preorder,结点,遍历,TreeNode,int,二叉树,null,重建 来源: https://www.cnblogs.com/lisen10/p/11042142.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。