原题链接:https://leetcode.com/problems/construct-binary-tree-from-inorder-and-postorder-traversal/
1. 题目介绍
Given inorder and postorder traversal of a tree, construct the binary tree.
给出一个树的中序遍历和后序遍历顺序,构造一棵二叉树
Note:
You may assume that duplicates do not exist in the tree.
你可以假设,树中没有重复的节点
For example, given
比如,分别给出中序遍历和后序遍历的路径:
inorder = [9,3,15,20,7]
postorder = [9,15,7,20,3]
Return the following binary tree:
需要返回下面这样的二叉树:
类似题目
105. Construct Binary Tree from Preorder and Inorder Traversal
给出一棵树的前序遍历和中序遍历,根据这两种遍历顺序构造一棵二叉树。
2. 解题思路
方法1 递归
-
在后序遍历中找到根节点
后序遍历的最后一个数必定是根节点,因此我们可以轻松知道根节点是谁。 -
在中序遍历中找到根节点
由题目可知,树中没有重复的节点,于是我们可以在中序遍历中找到根节点。 -
在中序遍历中找到左右子树的区间
中序遍历中,根节点之前的部分就是左子树,根节点之后的树就是右子树。同时可以计算出左子树的节点个数。 -
在后序遍历中找到左右子树的区间
根据左子树的长度,我们回到后序遍历,又可以找出在后序遍历中,左子树和右子树的区间在哪里。 -
递归重复上述4步,每一步都返回当前的根节点和左右子树。左右子树交给下一层递归完成计算。
实现代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode buildTree(int[] inorder, int[] postorder) {
return helper(inorder,0, inorder.length-1 , postorder, 0, postorder.length-1);
}
public TreeNode helper(int[] inorder , int in_start, int in_end, int[] postorder, int po_start, int po_end ){
if(in_start > in_end || po_start > po_end){
return null;
}
int rootVal = postorder[po_end];//后序遍历的最后一个值必定是根节点
//根据根节点的值,在中序遍历顺序中找到根节点的下标,存放于rootIndex
int rootIndex = 0;
for(int i = in_end; i >= in_start ; i--){
if(inorder[i] == rootVal){
rootIndex = i;
break;
}
}
//在中序遍历数组中,计算左子树的节点个数
int length = rootIndex - in_start;
TreeNode root = new TreeNode(rootVal);
root.left = helper(inorder, in_start, rootIndex-1, postorder, po_start, po_start+length-1);
//左子树的区间,在中序遍历的数组中,是[in_start, rootIndex-1],
//在后序遍历的数组中,是[po_start, po_start+length-1]
root.right= helper(inorder, rootIndex+1, in_end, postorder, po_start+length,po_end-1);
//右子树的区间,在中序遍历的数组中,是[rootIndex+1, in_end],
//在后序遍历的数组中,是[po_start+length, po_end-1]
return root;
}
}
方法2 递归+HashMap
是对方法1的改进,使用HashMap存放中序遍历的数值和下标,当需要求根节点的下标时就可以利用hashMap找出来。
实现方法
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int [] in ;
int [] po ;
HashMap <Integer, Integer> map = new HashMap<>();
int po_rootIndex ;
public TreeNode buildTree(int[] inorder, int[] postorder) {
in = inorder;
po = postorder;
po_rootIndex = postorder.length - 1;
int Count = 0;
for(int i : inorder){
map.put(i , Count);
Count++;
}
return helper(0, in.length-1);
}
public TreeNode helper(int l , int r){
if(l > r){
return null;
}
int rootVal = po[po_rootIndex];//在后序遍历中找到根节点的值
po_rootIndex --;
int in_rootIndex = map.get(rootVal);//在中序遍历中找到根节点的下标
TreeNode root = new TreeNode(rootVal);
//此处必须先构建右子树,再构建左子树
//因为后序遍历的数组,根节点的下标向前是右子树,再向前才是左子树
root.right = helper(in_rootIndex+1 , r);
root.left = helper(l , in_rootIndex-1);
return root;
}
}