第4题

题目:

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

例如:

前序遍历序列{1,2,4,7,3,5,6,8}

中序遍历序列{4,7,2,1,5,3,8,6}

重建二叉树并输出它的头节点

 

思路:

1.由前序遍历的第一个节点可知根节点为1,由此可以把中序遍历分为两部分,左边部分为根节点的左子树{4,7,2}(inOrder),右边部分为根节点的右子树{5,3,8,6}(inOrder)

2.由{4,7,2}(inOrder)与前序遍历对比知其前序遍历为{2,4,7}(preOrder)

 可知2为左子树第一级的“根节点”,其左子树为{4,7}(inOrder),其右子树为空,其前序遍历为{4,7}(preOrder)

 可知4为左子树第二级的“根节点”,其左子树为空,右子树为{7}

 到此,左子树遍历完成,右子树同理

3.将以上思路递归

 

代码实现

public class Third {
   class TreeNode{
       int val;
       TreeNode left;
       TreeNode right;
       public TreeNode(int val){
           this.val = val;
       }
   }

   public void reConstructBST(int[] pre, int[] in){
        reBuildBST(pre, 0, pre.length - 1, in, 0, in.length - 1);
   }

   private TreeNode reBuildBST(int[] pre, int preStart, int preEnd, int[] in, int inStart, int inEnd){
       if(preStart > preEnd || inStart > inEnd){
           return null;
       }
       TreeNode root = new TreeNode(pre[preStart]);
       for(int i = 0; i < inEnd; i ++){
           if(in[i] == pre[preStart]){
               root.left = reBuildBST(pre, preStart, preStart + i - inStart, in, inStart, i - 1);
               root.right = reBuildBST(pre, preStart + i - inStart + 1, preEnd, in, i + 1, inEnd);
           }
       }
       return root;
   }
}

 

转载于:https://www.cnblogs.com/HarSong13/p/11324477.html

你可能感兴趣的:(第4题)