中序和后序遍历构造二叉树

题目链接:

涉及知识:二叉树的遍历

分析:

上一篇中介绍了如何通过二叉树的前序和中序遍历构造二叉树。

我们知道前序的遍历顺序是:根,左,右;中序的遍历顺序是左,根,右;后序的遍历顺序是左,右,根;

如果我们将后序遍历倒过来看便是根,右,左;会发现和前序遍历是非常相似的。前序遍历依次是根节点,左子树根节点,右子树根节点;后序遍历倒过来依次是根节点,右子树根节点,左子树根节点;因此解法和前序+中序生成后序的思路一样,从后序遍历中倒过来查找根节点,且先生成该节点的右子树,在生成该节点的左子树即可。

代码:

import java.util.HashMap;

/*
 * @lc app=leetcode.cn id=106 lang=java
 *
 * [106] 从中序与后序遍历序列构造二叉树
 */
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    private int[] inorder;
    private int[] postorder;
    private HashMap<Integer, Integer> map = new HashMap<>();
    private int tag;

    public TreeNode buildTree(int[] inorder, int[] postorder) {
        this.inorder = inorder;
        this.postorder = postorder;
        this.tag = postorder.length - 1;

        for(int i = 0; i < inorder.length; ++i){
            map.put(inorder[i], i);
        }

        return generateTree(0, postorder.length - 1);
    }

    //通过中序遍历的 s 到 e 生成二叉树,并返回根节点
    public TreeNode generateTree(int s, int e){
        if(s > e){
            return null;
        }

        int index = map.get(postorder[tag]);
        //构造根节点
        TreeNode node = new TreeNode(postorder[tag--]);
        //生成右子树
        node.right = generateTree(index + 1, e);
        //生成左子树
        node.left = generateTree(s, index - 1);

        return node;
    }
}
posted @ 2019-09-06 12:17  telankesi  阅读(1333)  评论(0编辑  收藏  举报