（重建二叉树）从前序与中序遍历序列构造二叉树

最新推荐文章于 2024-07-02 23:33:14 发布

原创最新推荐文章于 2024-07-02 23:33:14 发布 · 286 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Tree

算法刷题专栏收录该内容

6 篇文章

订阅专栏

本文介绍如何根据给定的前序遍历与中序遍历序列构造二叉树的方法。通过递归方式确定根节点，并划分左右子树进行构建。

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如，给出

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

返回如下的二叉树：

分析：先序遍历的顺序是（根-左子树-右子树），中序遍历的顺序是（左子树-根-右子树），所以先序遍历输出的第一个节点一定是根节点，从而分别递归构造子树。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        int length1=preorder.length-1;
        int length2=inorder.length-1;
        return rebuildTree(preorder,0,length1,inorder,0,length2);
    }
    public TreeNode rebuildTree(int[] preorder,int start1,int end1,int[] inorder,int start2,int end2){
       if(start1>end1||start2>end2)
               return null;
       TreeNode root=new TreeNode(preorder[start1]);
       for(int i=start2;i<=end2;i++){
           if(inorder[i]==preorder[start1]){
               root.left=rebuildTree(preorder,start1+1,start1+i-start2,inorder,start2,i-1);
               root.right=rebuildTree(preorder,start1+i-start2+1,end1,inorder,i+1,end2);
               break;
           }
       }
        return root;
    } 
}