【LeetCode-面试算法经典-Java实现】【106-Construct Binary Tree from Inorder and Postorder Traversal（构造二叉树II）】

【106-Construct Binary Tree from Inorder and Postorder Traversal（通过中序和后序遍历构造二叉树）】

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

原题

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

题目大意

　　给定一个中序遍历和后序遍历序列，构造一棵二叉树
　　注意：
　　树中没有重复元素

解题思路

　　后序遍历的最后一个元素就是树的根结点(值为r)，在中序遍历的序列中找值为r的位置idx，idx将中序遍历序列分为左右两个子树，对应可以将后序遍历的序列分在两个子树，递归对其进行求解。

代码实现

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

算法实现类

public class Solution {

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

        // 参数检验
        if (inorder == null || postorder == null || inorder.length == 0
                || inorder.length != postorder.length) {
            return null;
        }

        // 构建二叉树
        return solve(inorder, 0, inorder.length - 1, postorder, 0, postorder.length - 1);
    }

    /** * 构建二叉树 * * @param inorder 中序遍历的结果 * @param x 中序遍历的开始位置 * @param y 中序遍历的结束位置 * @param postorder 后序遍历的结果 * @param i 后序遍历的开始位置 * @param j 后序遍历的结束位置 * @return 二叉树 */
    public TreeNode solve(int[] inorder, int x, int y, int[] postorder, int i, int j) {

        if (x >= 0 && x <= y && i >= 0 && i <= j) {
            // 只有一个元素，（此时也有i=j成）
            if (x == y) {
                return new TreeNode(postorder[j]);
            }
            // 多于一个元素，此时也有i<j
            else if (x < y) {
                // 创建根结点
                TreeNode root = new TreeNode(postorder[j]);

                // 找根结点在中序遍历的下标
                int idx = x;
                while (idx < y && inorder[idx] != postorder[j]) {
                    idx++;
                }

                // 左子树非空，构建左子树
                int leftLength = idx - x;
                if (leftLength > 0) {
                    // i, i + leftLength - 1，前序遍历的左子树的起始，结束位置
                    root.left = solve(inorder, x, idx - 1, postorder, i, i + leftLength - 1);
                }

                // 右子树非空，构建右子树
                int rightLength = y - idx;
                if (rightLength > 0) {
                    // i + leftLength, j - 1，前序遍历的右子树的起始，结束位置
                    root.right = solve(inorder, idx + 1, y, postorder, i + leftLength, j - 1);
                }

                return root;
            } else {
                return null;
            }
        }

        return null;
    }
}

评测结果

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

特别说明

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