通用方式解决根据二叉树遍历重建二叉树问题
重做LeetCode题目的时候,发现两道题目非常类似,但之前却用了两种写法差别比较大的解法。感觉可以合并为同一种方式进行解决,于是找出了如下的通用方式解决这两个问题
题目一:
输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
题目二:
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
可以看出仅仅是给出的两种遍历方式的不同。
大家都知道根据 前序+中序遍历 或 中序+后序遍历可以唯一确定一棵二叉树
前序遍历:根 左 右
中序遍历:左 根 右
后序遍历:左 右 根
这两道题的思路都非常类似:
根据前序遍历或后序遍历确定出根节点
在从中序遍历中获取其左右子树
递归上述步骤即可
题目一代码:
class Solution {
private int preorder_index;
private int[] inorder;
private int[] preorder;
private Map<Integer,Integer> idx_map;
public TreeNode buildTree(int[] preorder, int[] inorder) {
this.inorder = inorder;
this.preorder = preorder;
this.preorder_index = 0;
this.idx_map = new HashMap<>();
for (int i = 0; i < inorder.length; i++) {
idx_map.put(inorder[i],i);
}
return buildTree(0,preorder.length-1);
}
public TreeNode buildTree(int left,int right){
if (left > right)
return null;
TreeNode root = new TreeNode(preorder[preorder_index]);
Integer in_index = idx_map.get(root.val);
this.preorder_index++;
root.left = buildTree(left,in_index-1);
root.right = buildTree(in_index+1,right);
return root;
}
}
题目二代码:
class Solution {
private int postorder_index;
private int[] inorder;
private int[] postorder;
private Map<Integer,Integer> idx_map;
public TreeNode buildTree(int[] inorder, int[] postorder) {
//初始化成员变量
this.inorder = inorder;
this.postorder = postorder;
this.idx_map = new HashMap<>();
this.postorder_index = postorder.length - 1; //从后序遍历的最后一个元素开始
//将中序遍历的 元素 - 下标 作为K V 放入map集合
for (int i = 0; i < inorder.length; i++) {
this.idx_map.put(inorder[i],i);
}
TreeNode root = handler(0, postorder.length - 1);
return root;
}
/**
* 构建树的处理方法
* @param left 后序遍历的左指针
* @param right 后序遍历的右指针
* @return
*/
public TreeNode handler(int left,int right){
//当左指针 > 右指针代表已经没有结点元素 可以用来构建树 直接返回null
if(left > right){
return null;
}
//有元素的情况下 取出后序遍历的最后一个元素作为跟结点
TreeNode node = new TreeNode(postorder[postorder_index]);
//找到根结点在中序遍历中的位置 , 将中序遍历结果分为左右两棵子树
int index = idx_map.get(postorder[postorder_index]);
//将后序遍历长度-1 取出当前的元素结点
postorder_index --;
node.right = handler(index+1,right);
node.left = handler(left,index-1);
return node;
}
}
代码差距只有一行 找找看