Day55 从前序与中序遍历序列构造二叉树

最新推荐文章于 2022-08-27 12:43:03 发布

微光长明

最新推荐文章于 2022-08-27 12:43:03 发布

阅读量144

点赞数 1

分类专栏： LeetCode刷题集文章标签：二叉树 leetcode 算法数据结构

本文链接：https://blog.csdn.net/walkmanfy/article/details/115093673

版权

LeetCode刷题集专栏收录该内容

90 篇文章 0 订阅

订阅专栏

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

https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/

示例1：

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

例如，给出

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

返回如下的二叉树：
    3
   / \
  9  20
    /  \
   15   7

Java解法

思路：

前序遍历，第一个为根节点，第二个为左节点== [ 根节点, [左子树的前序遍历结果], [右子树的前序遍历结果] ]
中序遍历，第一个为最左节点，第二个为最左节点父节点 == [ [左子树的中序遍历结果], 根节点, [右子树的中序遍历结果] ]
因为没有重复元素，在遍历到值相同时认为是同一节点,采用递归处理

package sj.shimmer.algorithm.m3_2021;

import java.util.Arrays;

import sj.shimmer.algorithm.TreeNode;

/**
 * Created by SJ on 2021/3/21.
 */

class D55 {
    public static void main(String[] args) {
        TreeNode root = buildTree(new int[]{3, 9, 20, 15, 7}, new int[]{9, 3, 15, 20, 7});
        System.out.println(root);
    }
    public static TreeNode buildTree(int[] preorder, int[] inorder) {
        //- 前序遍历[ 根节点, [左子树的前序遍历结果], [右子树的前序遍历结果] ]
        //- 中序遍历[ [左子树的中序遍历结果], 根节点, [右子树的中序遍历结果] ]
        if (preorder == null||preorder.length==0||inorder==null||inorder.length==0) {
            return null;
        }
        int length = inorder.length;
        for (int i = 0; i < length; i++) {
            if (inorder[i]==preorder[0]) {
                TreeNode root = new TreeNode(inorder[i]);
                //左子树长度
                int left = i;
                //前序 左子树
                int[] tempLeftP = Arrays.copyOfRange(preorder, 1, left+1);
                int[] tempLeftI = Arrays.copyOfRange(inorder, 0, left);
                root.left = buildTree(tempLeftP, tempLeftI);
                //前序 右子树
                int[] tempRightP = Arrays.copyOfRange(preorder, left+1, length);
                int[] tempRightI = Arrays.copyOfRange(inorder, left+1, length);
                root.right = buildTree(tempRightP, tempRightI);
                return root;
            }
        }
        return null;
    }
}