剑指 Offer 07. 重建二叉树

题目:

输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

例如,给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树:

    3
   / \
  9  20
    /  \
   15   7

限制:

0 <= 节点个数 <= 5000

难度:中等

代码:

//前序根-左-右,中序左-根-右,可以看出从前序取出一个值,再在中序找到其位置,可以发现,中序位置左侧是其左子树,右侧是右子树。通过中序可以得到其左右子树的长度,根据长度

//可以在递归时确定先序遍历数组左、右子树的下标位置,以及中序数组左、右子树的下标位置,从而可以确定递归方法的参数。

 1 /**
 2  * Definition for a binary tree node.
 3  * public class TreeNode {
 4  *     int val;
 5  *     TreeNode left;
 6  *     TreeNode right;
 7  *     TreeNode(int x) { val = x; }
 8  * }
 9  */
10 class Solution {
11     public TreeNode buildTree(int[] preorder, int[] inorder) {
12         if(preorder.length==0){return null;}
13         if(preorder.length==1){return new TreeNode(preorder[0]);}  //空树和只有根节点树单独处理,可以提高执行效率
14         return fun1(preorder,0,preorder.length-1,inorder,0,inorder.length-1);
15     }
16 
17     public static TreeNode fun1(int[] preorder,int leftp,int rightp, int[] inorder,int lefti,int righti){
18         if(leftp<=rightp){  //递归终止条件
19             int rootNum=preorder[leftp]; //当前树的根的值
20             int t=0;
21             for(int i=lefti;i<=righti;i++){
22                 if(inorder[i]==rootNum){
23                     t=i;
24                 }
25             }
26 
27             int len1=t-lefti;  //左子树的长度
28             int len2=righti-t; //右子树的长度                                         
29             TreeNode root=new TreeNode(rootNum);
30             root.left=fun1(preorder,leftp+1,leftp+len1,inorder,lefti,t-1);  //循环左子树
31             root.right=fun1(preorder,rightp-len2+1,rightp,inorder,t+1,righti);  //循环右子树
32             return root;
33         }
34         return null;
35     }
36 }

剑指 Offer 07. 重建二叉树

 

上一篇:vb.net 教程 8-3 数据库操作3


下一篇:面试题7:重建二叉树