一、题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如,给出
前序遍历 preorder = [3,9,20,15,7] 中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3 / \ 9 20 / \ 15 7
题目来源:二叉树重建
二、解题思路
二叉树前序遍历的顺序为[根节点|左子树|右子树]
二叉数中序遍历的顺序为[左子树|根节点|右子树]
由此可知:
前序遍历的首个元素即为根节点
根据前序的根节点,把中序遍历划分为[左子树|根|右子树]
根据中序遍历左(右)子树节点的数量,可以将前序遍历划分为[根节点|左子树|右子树]
确定了三个节点的关系,根节点,左子树的根,右子树的根
由二叉树的定义,本题可以采用递归的方式来解决。
递归的参数为:前序遍历的根节点索引 rootIndex,中序遍历的左边界inorderLeft,中序遍历的右边界inorderRight
递归结束条件:inorderLeft>inorderRight
递归过程:
1、先创建根节点
2、得到根节点在中序遍历中的索引 i
3、构建根节点的左子树和右子树
左子树的根为rootIndex+1,中序遍历的左边界为inorderLeft,中序遍历的右边界i-1
右子树的根为rootIndex+i-inordereft+1,中序遍历左边界 i+1,中序遍历右边界inorderRight
代码如下:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
//重建二叉树 递归的方式
import java.util.*;
class Solution {
private int []preorder;
private Map<Integer,Integer> dict=new HashMap<Integer,Integer>();
public TreeNode buildTree(int[] preorder, int[] inorder) {
this.preorder=preorder;
for (int i=0;i<inorder.length;i++){
dict.put(inorder[i],i);
}
return helper(0,0,inorder.length-1);
}
private TreeNode helper(int rootIndex,int inorderLeft,int inorderRight){
if(inorderLeft>inorderRight){
return null;
}
TreeNode root=new TreeNode(preorder[rootIndex]);
int i=dict.get(preorder[rootIndex]);
root.left=helper(rootIndex+1,inorderLeft,i-1);
root.right=helper(rootIndex+i-inorderLeft+1,i+1,inorderRight);
return root;
}
}