输入某二叉树的前序遍历和中序遍历的结果,请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
解题思路
前序遍历:根左右
中序遍历:左根右
前序遍历第一个值即为当前子树根节点的值
根据前序遍历根节点的值,将中序遍历划分为左右两个部分,并以此递归,重建该二叉树
代码
/**
* 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[] preorder, int[] inorder) {
int pl = 0 ,pr = preorder.length-1;
int il = 0, ir = inorder.length-1;
return help(preorder,pl,pr,inorder,il,ir);
}
public TreeNode help(int[] preorder,int pl,int pr,int[] inorder,int il ,int ir){
if(pl > pr || il > ir) return null;
int i = il;
while(preorder[pl] != inorder[i]){
i++;
}
TreeNode root =new TreeNode(inorder[i]);
//i-il,寻找到当前根节点的距离,即下一次递归,节点所需要的移动的距离
root.left = help(preorder,pl+1,pl+i-il,inorder,il,i-1);
root.right = help(preorder,pl+i+1-il,pr,inorder,i+1,ir);
return root;
}
}
作者:mazyg
链接:https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof/solution/di-gui-qiu-jie-by-mazyg-d2ir/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。