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

本文最后更新于 628 天前，其中的信息可能已经有所发展或是发生改变。

1. 要点

利用Map来优化For循环，减少时间复杂度

2. 题目

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

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

例如，给出

前序遍历 preorder = [3,9,20,15,7] 中序遍历 inorder = [9,3,15,20,7] 返回如下的二叉树：

代码语言：javascript复制

3. 代码

代码语言：javascript复制

package com.yuyy.java.training.base.algorithm;

import org.junit.Test;

import java.util.HashMap;
import java.util.Map;

public class 从前序与中序遍历序列构造二叉树 {

    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }
    }

    private Map<Integer, Integer> preorderMap;

    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if (preorder.length == 0) {
            return null;
        }
        preorderMap = new HashMap();
        for (int j = 0; j < preorder.length; j  ) {
            preorderMap.put(preorder[j], j);
        }
        return assemblyTree(inorder);
    }

    public int findRoot(int[] inorder) {
        int min = Integer.MAX_VALUE;
        int root=0;
        for (int i = 0; i < inorder.length; i  ) {
            if (min>preorderMap.get(inorder[i])){
                min=preorderMap.get(inorder[i]);
                root=i;
            }
        }
        return root;
    }

    public TreeNode assemblyTree(int[] inorder) {
        int root = findRoot(inorder);
        TreeNode node = new TreeNode(inorder[root]);
        int[] leftArr = new int[root];
        int[] rightArr = new int[inorder.length - root - 1];
        int k = 0;
        for (int i = 0; i < inorder.length; i  ) {
            if (i < root) {
                leftArr[i] = inorder[i];
            }
            if (i > root) {
                rightArr[k  ] = inorder[i];
            }
        }
        if (root > 0) {
            node.left = assemblyTree(leftArr);
        }
        if (root < inorder.length - 1) {
            node.right = assemblyTree(rightArr);
        }
        return node;
    }

    @Test
    public void test() {
        int[] preorder = new int[]{3, 9, 20, 15, 7};
        int[] inorder = new int[]{9, 3, 15, 20, 7};
        TreeNode treeNode = buildTree(preorder, inorder);
        System.out.println(treeNode);
    }
}

Post Views: 328

编程算法二叉树

0 人点赞