思路 根据两个顺序构造一个唯一的二叉树,以 后先序数组的第一个元素为切割点,先切中序数组,根据中序数组,反过来在切先序数组。一层一层切下去,每次先序数组第一个元素就是节点元素。
第一步:如果数组大小为零的话,说明是空节点了。
第二步:如果不为空,那么取先序数组第一个元素作为节点元素。
第三步:找到先序数组第一个元素在中序数组的位置,作为切割点
第四步:切割中序数组,切成中序左数组和中序右数组 (顺序别搞反了,一定是先切中序数组)
第五步:切割先序数组,切成先序左数组和先序右数组
第六步:递归处理左区间和右区间
难点就是如何切割,以及边界值找不好很容易乱套。
首先要切割中序数组,为什么先切割中序数组呢?
切割点在先序数组的第一个元素,就是用这个元素来切割中序数组的,所以必要先切割中序数组。
中序数组相对比较好切,找到切割点(先序数组的第一个元素)在中序数组的位置,然后切割,如下代码中我坚持左闭右开的原则:
int rootIndex=map.get(preorder[preBegin]);//找到前序遍历第一个元素在中序中的位置
接下来切割先序数组。
首先后序数组的第一个元素指定不能要了,这是切割点 也是 当前二叉树中间节点的元素,已经用了。
先序数组的切割点怎么找?
先序数组没有明确的切割元素来进行左右切割,不像中序数组有明确的切割点,切割点左右分开就可以了。
此时有一个很重的点,就是中序数组大小一定是和先序数组的大小相同的(这是必然)。
中序数组我们都切成了左中序数组和右中序数组了,那么先序数组就可以按照左中序数组的大小来切割,切成左先序数组和右先序数组。
/*** 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 {Map<Integer,Integer> map;public TreeNode buildTree(int[] preorder, int[] inorder) {map=new HashMap<>();for(int i=0;i<inorder.length;i++){map.put(inorder[i],i);}return findNode(preorder,0,preorder.length,inorder,0,inorder.length);}public TreeNode findNode(int[] preorder,int preBegin,int preEnd,int[] inorder,int inBegin,int inEnd){if(preBegin>=preEnd||inBegin>=inEnd){//没有元素,返回空return null;}int rootIndex=map.get(preorder[preBegin]);//找到前序遍历第一个元素在中序中的位置TreeNode root=new TreeNode(inorder[rootIndex]);//构造结点int lenOfLeft=rootIndex-inBegin;//保存中序左子树个数root.left=findNode(preorder,preBegin+1,preBegin+lenOfLeft+1,inorder,inBegin,rootIndex);root.right=findNode(preorder,preBegin+lenOfLeft+1,preEnd,inorder,rootIndex+1,inEnd);return root;}
}