public TreeNode buildTree(int[] preorder, int[] inorder) {
return construct(0, preorder.length - 1, preorder, 0, inorder.length - 1, inorder);
}
public TreeNode construct(int preStart, int preEnd, int[] preorder, int inStart, int inEnd, int[] inorder) {
if(inStart > inEnd || preStart > preEnd) {
return null;
}
TreeNode root = new TreeNode(preorder[preStart]);
int i = inStart;
for(; i <= inEnd; i++) {
if(inorder[i] == preorder[preStart]) {
break;
}
}
root.left = construct(preStart + 1, preStart + i - inStart, preorder, inStart, i - 1, inorder); //i - inStart是左子树的长度
root.right = construct(preStart + i - inStart + 1, preEnd, preorder, i + 1, inEnd, inorder);
return root;
}
这种构造一颗二叉树的题目,一般是以递归的方式生成的,先生成父节点,再递归生成左子树和右子树。
前序遍历的数组中,第一个下标的值即为根节点的值;
在中序遍历的数组中找到这个值,并记录下标;
则此下标往左即为左子树的部分,下标往右即为右子树的部分;
算出左子树部分的长度,在前序遍历数组中第一个下标加上这个长度即为左子树部分,剩下的部分则为右子树;
再分别递归生成左右子树(注意:这个长度很重要!!!)。
类似的题目还有:
106. Construct Binary Tree from Inorder and Postorder Traversal
public TreeNode buildTree(int[] inorder, int[] postorder) {
return construct(postorder, 0, postorder.length - 1, inorder, 0, inorder.length - 1);
}
public TreeNode construct(int[] postorder, int postStart, int postEnd, int[] inorder, int inStart, int inEnd) {
if(postStart > postEnd || inStart > inEnd) {
return null;
}
TreeNode root = new TreeNode(postorder[postEnd]);
int i = inStart;
for(; i <= inEnd; i++) {
if(inorder[i] == root.val) {
break;
}
}
root.right = construct(postorder, postEnd - inEnd + i, postEnd - 1, inorder, i + 1, inEnd); //inEnd - i是右子树的长度
root.left = construct(postorder, postStart, postEnd - inEnd + i - 1, inorder, inStart, i - 1);
return root;
}
108. Convert Sorted Array to Binary Search Tree
public TreeNode sortedArrayToBST(int[] nums) {
return construct(0, nums.length - 1, nums);
}
public TreeNode construct(int begin, int end, int[] nums) {
if(begin > end) {
return null;
}
int mid = (begin + end + 1) / 2;
TreeNode root = new TreeNode(nums[mid]);
root.left = construct(begin, mid - 1, nums);
root.right = construct(mid + 1, end, nums);
return root;
}
654. Maximum Binary Tree
public TreeNode constructMaximumBinaryTree(int[] nums) {
return construct(0, nums.length - 1, nums);
}
public TreeNode construct(int begin, int end, int[] nums) {
if(begin > end) {
return null;
}
int indexOfMax = begin;
int max = nums[begin];
for(int i = begin; i <= end; i++) {
if(nums[i] > max) {
indexOfMax = i;
max = nums[i];
}
}
TreeNode root = new TreeNode(max);
root.left = construct(begin, indexOfMax - 1, nums);
root.right = construct(indexOfMax + 1, end, nums);
return root;
}
这几道题都是给定数组,让你生成一颗二叉树的题
相似之处在于:
1.找到父节点的值(根据题目要求,或者某些性质);
2.把数组分割成两部分,一部分是左子树,另一部分是右子树(可能需要计算长度);
3.利用这两部分递归生成左右子树