Construct Binary Tree from Preorder and Inorder Traversal 根据前序遍历和中序遍历构造二叉树

来源:互联网 发布:java 时间戳转换成时间 编辑:程序博客网 时间:2024/05/29 07:43

Given preorder and inorder traversal of a tree, construct the binary tree.

Note:
You may assume that duplicates do not exist in the tree.

假设我们的二叉树长这样。

它的前序遍历为:4,7,9,10,8

它的中序遍历为:9,7,10,4,8

我们可以看到前序遍历的第一个节点4, 正是root。

看看4在中序遍历中的位置:9,7,10,4,8

4将中序遍历分成了左子树节点右子树节点两部分。

我们再看看4节点的左子树:

9,7,10前序遍历的下一个值7, 又是这个序列的root。

又将子树分成9, 10 两部分。

前序遍历的下一个值9,对于7节点左子树序列 9,(因为左子树序列只有9了,没有左右子树了)。

前序遍历中的值负责对中序遍历节点值进行左右子树划分。

这便是我们递归建树的思想。

先看未优化的运行时间:


代码:

    private int preIndex = 0;    public TreeNode buildTree(int[] preorder, int[] inorder) {        return doBuildTree(preorder, inorder, 0, inorder.length - 1);    }    public TreeNode doBuildTree(int[] preorder, int[] inorder, int inBeigin, int inEnd) {        if (inBeigin > inEnd) {            return null;        }        int mid = finIndex(inorder, inBeigin, inEnd, preorder[preIndex]);        TreeNode root = new TreeNode(preorder[preIndex]);        preIndex++;        root.left = doBuildTree(preorder, inorder, inBeigin, mid - 1);        root.right = doBuildTree(preorder, inorder, mid + 1, inEnd);        return root;    }    public int finIndex(int[] arr, int begin, int end, int val) {        for (int i = begin; i <= end; i++) {            if (arr[i] == val) {                return i;            }        }        return -1; // will not happen    }

之前查找前序遍历中的值在中序遍历的下标,是进行线性查找。

现在我们将值与下标的信息放到map中,使查找的时间复杂度从 O (n) 变成 O (1)。

运行时间:


代码:

public class ConstructBinaryTreefromPreorderandInorderTraversal {    private int preIndex = 0;    private Map<Integer, Integer> valueToIndex = new HashMap<>();    public TreeNode buildTree(int[] preorder, int[] inorder) {        for (int i = 0; i < inorder.length; i++) {            valueToIndex.put(inorder[i], i);        }        return doBuildTree(preorder, inorder, 0, inorder.length - 1);    }    public TreeNode doBuildTree(int[] preorder, int[] inorder, int inBeigin, int inEnd) {        if (inBeigin > inEnd) {            return null;        }        int mid = valueToIndex.get(preorder[preIndex]);        TreeNode root = new TreeNode(preorder[preIndex]);        preIndex++;        root.left = doBuildTree(preorder, inorder, inBeigin, mid - 1);        root.right = doBuildTree(preorder, inorder, mid + 1, inEnd);        return root;    }}






1 0
原创粉丝点击