涉及知识:二叉树的遍历
分析:
上一篇中介绍了如何通过二叉树的前序和中序遍历构造二叉树。
我们知道前序的遍历顺序是:根,左,右;中序的遍历顺序是左,根,右;后序的遍历顺序是左,右,根;
如果我们将后序遍历倒过来看便是根,右,左;会发现和前序遍历是非常相似的。前序遍历依次是根节点,左子树根节点,右子树根节点;后序遍历倒过来依次是根节点,右子树根节点,左子树根节点;因此解法和前序+中序生成后序的思路一样,从后序遍历中倒过来查找根节点,且先生成该节点的右子树,在生成该节点的左子树即可。
代码:
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;
}
}