码迷,mamicode.com
首页 > 编程语言 > 详细

【LeetCode-面试算法经典-Java实现】【105-Construct Binary Tree from Preorder and Inorder Traversal(构造二叉树)】

时间:2015-08-09 07:13:59      阅读:234      评论:0      收藏:0      [点我收藏+]

标签:前序   后序   二叉树   遍历   算法   

【105-Construct Binary Tree from Preorder and Inorder Traversal(通过前序和中序遍历构造二叉树)】


【LeetCode-面试算法经典-Java实现】【所有题目目录索引】

原题

  Given preorder and inorder traversal of a tree, construct the binary tree.
  Note:
  You may assume that duplicates do not exist in the tree.

题目大意

  给定一个前序和中序遍历序列,构建一个二叉树
  注意:
   - 二叉树中元素重复元素

解题思路

  前序遍历第一个元素是根结点(k),在中序遍历序列中找值为k的下标idx,idx将中序遍历序列分成左右子树,对前序遍历序列也一样,可进行递归操作

代码实现

树结点类

public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

算法实现类一:

public class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if (preorder == null || preorder.length ==0){
            return null;
        }
        HashMap<Integer, Integer> inorderMap = new HashMap<Integer, Integer>();
        for (int i=0;i<inorder.length;i++){
            inorderMap.put(inorder[i],i);
        }

        Deque<TreeNode> stack = new LinkedList<TreeNode>();
        TreeNode root = new TreeNode(preorder[0]);
        stack.push(root);
        for (int i=1;i<preorder.length;i++){
            TreeNode top = stack.peek();
            int indexTop = inorderMap.get(top.val);
            int indexVal = inorderMap.get(preorder[i]);

            TreeNode node = new TreeNode(preorder[i]);

            if (indexVal<indexTop){
                top.left=node;
            }
            else{
            while (indexVal>indexTop){
                    top = stack.pop();
                    indexTop = stack.isEmpty()?Integer.MAX_VALUE:inorderMap.get(stack.peek().val);
                }
                top.right = node;
            }
            stack.push(node);
        }
        return root;
    }
}

算法实现类二:(会超时)

public class Solution {

    public TreeNode buildTree(int[] preorder, int[] inorder) {

        // 参数校验
        if (preorder == null || inorder == null || preorder.length == 0
                || preorder.length != inorder.length) {
            return null;
        }
        return solve(preorder, 0, preorder.length - 1, inorder, 0, inorder.length - 1);
    }

    /**
     * 构建二叉树,数据输入的正确性由输入数据自己保证
     *
     * @param preorder 先序遍历的结果
     * @param x        先序遍历的开始位置
     * @param y        先序遍历的结束位置
     * @param inorder  中序遍历的结果
     * @param i        中序遍历的开始位置
     * @param j        中序遍历的结束位置
     * @return 二叉树的根结点
     */
    public TreeNode solve(int[] preorder, int x, int y, int[] inorder, int i, int j) {

        if (x >= 0 && x <= y && i >= 0 && i <= j) {
            // 只有一个元素
            if (x == y) {
                return new TreeNode(preorder[x]);
            } else if (x < y) {
                // 记录根结点的索引
                int idx = i;
                while (idx <= j && inorder[idx] != preorder[x]) {
                    idx++;
                }

                // 创建根结点
                TreeNode root = new TreeNode(inorder[idx]);

                // 左子树的结点个数
                int leftLength = idx - i;
                //
                if (leftLength > 0) {
                    // x + 1, x + leftLength:左子树起始和结束位置
                    root.left = solve(preorder, x + 1, x + leftLength, inorder, i, idx - 1);
                }

                // 右子树的结点个数
                int rightLength = j - idx;
                if (rightLength > 0) {
                    // x + leftLength + 1, y:右子树起始和结束位置
                    root.right = solve(preorder, x + leftLength + 1, y, inorder, idx + 1, j);
                }
                return root;
            }
        }

        return null;
    }
}

评测结果

  点击图片,鼠标不释放,拖动一段位置,释放后在新的窗口中查看完整图片。

技术分享

特别说明

欢迎转载,转载请注明出处【http://blog.csdn.net/derrantcm/article/details/47371985

版权声明:本文为博主原创文章,未经博主允许不得转载。

【LeetCode-面试算法经典-Java实现】【105-Construct Binary Tree from Preorder and Inorder Traversal(构造二叉树)】

标签:前序   后序   二叉树   遍历   算法   

原文地址:http://blog.csdn.net/derrantcm/article/details/47371985

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!