在leetcode上有两题,分别是根据前序/后序 和中序遍历重构二叉树。思路都是一样的,但是要记住,在从后序遍历重构二叉树的时候,要先从右子树进行建立,因为后序遍历先遍历的是右子树。
前序遍历+中序遍历
(1)首先进行边界判断,如果遍历数组为0或者1的情况下,说明这个树只有一个顶点或者没有顶点,直接进行构造。
(2)根据前序遍历和中序遍历建立二叉树的思路是:前序遍历依次每一个节点对应一个子树,这个节点在中序遍历中找到它相应的位置,它的左边(当然是在子树数组范围内)就是左子树的顶点,右边就是右子树的顶点。
前序遍历就是确定树的根节点,中序遍历就是确定树的左右子树。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int preIndex = 0;
int[] preorder;
int[] inorder;
Map<Integer, Integer> map = new HashMap<>();
public TreeNode buildTree(int[] preorder, int[] inorder) {
if(preorder.length == 0)
return null;
if(preorder.length == 1)
return new TreeNode(preorder[0]);
this.preorder = preorder;
this.inorder = inorder;
int len = inorder.length;
for(int i = 0; i < len; i++){
map.put(inorder[i], i);
}
TreeNode res = func(0, len-1);
return res;
}
public TreeNode func(int l, int r){
if(l > r)
return null;
TreeNode root = new TreeNode(preorder[preIndex]);
int inIndex = map.get(preorder[preIndex]);
preIndex++;
root.left = func(l, inIndex-1);
root.right = func(inIndex+1, r);
return root;
}
}
后序遍历+中序遍历
代码相同,注意的是需要先构建右子树。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
Map<Integer, Integer> map;
int postIndex;
public TreeNode buildTree(int[] inorder, int[] postorder) {
int len = inorder.length;
if(len == 0)
return null;
if(len == 1)
return new TreeNode(inorder[0]);
map = new HashMap<>();
postIndex = len-1;
for(int i = 0; i < len; i++)
{
map.put(inorder[i], i);
}
TreeNode res = fuc(0, len-1, inorder, postorder);
return res;
}
public TreeNode fuc(int l, int r, int[] inorder, int[] postorder){
if(l > r)
return null;
TreeNode newNode = new TreeNode(postorder[postIndex]);
System.out.println(newNode.val);
int inIndex = map.get(postorder[postIndex]);
postIndex--;
newNode.right = fuc(inIndex+1, r, inorder, postorder);
newNode.left = fuc(l, inIndex-1, inorder, postorder);
return newNode;
}
}