LeetCode 105. 从前序与中序遍历序列构造二叉树
@TOC
给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。
105. 从前序与中序遍历序列构造二叉树
示例:
提示:
1 <= preorder.length <= 3000
inorder.length == preorder.length
-3000 <= preorder[i], inorder[i] <= 3000
preorder 和 inorder 均 无重复 元素
inorder 均出现在 preorder
preorder 保证 为二叉树的前序遍历序列
inorder 保证 为二叉树的中序遍历序列
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
private Map indexMap;
public TreeNode buildTree(int[] preorder, int[] inorder) {
int pLen = preorder.length;
int iLen = inorder.length;
indexMap = new HashMap<>();
for(int i = 0; i < pLen;i++){
indexMap.put(inorder[i],i);
}
return dfs(preorder,inorder,0,pLen - 1,0,iLen - 1);
}
TreeNode dfs(int[] preorder, int[] inorder,int pl,int pr,int il,int ir){
//终止程序 条件是所有元素遍历完毕
if(pl > pr){
return null;
}
TreeNode nowNode = new TreeNode(preorder[pl]);
//pos 是根节点坐标
int pos = indexMap.get(preorder[pl]);
int cnt = pos - il;
//分别设置 先序,中序长度。不要人肉递归
nowNode.left = dfs(preorder,inorder,pl + 1,pl + cnt,il,pos - 1);
nowNode.right = dfs(preorder,inorder,pl + cnt + 1,pr,pos + 1,ir);
return nowNode;
}
}
递归只需要总结当前层的处理逻辑
留言与评论(共有 0 条评论) “” |