java先序遍历构造二叉树
时间: 2023-10-14 07:15:54 浏览: 99
以下是Java代码,使用递归实现先序遍历构造二叉树:
```java
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
int preLen = preorder.length;
int inLen = inorder.length;
// 构建二叉树
return buildTreeHelper(preorder, 0, preLen - 1, inorder, 0, inLen - 1);
}
private TreeNode buildTreeHelper(int[] preorder, int preStart, int preEnd, int[] inorder, int inStart, int inEnd) {
if (preStart > preEnd || inStart > inEnd) {
// 递归终止条件
return null;
}
// 先序遍历的第一个节点为根节点
int rootVal = preorder[preStart];
TreeNode root = new TreeNode(rootVal);
// 在中序遍历中找到根节点的位置
int rootIndex = 0;
for (int i = inStart; i <= inEnd; i++) {
if (inorder[i] == rootVal) {
rootIndex = i;
break;
}
}
// 递归构建左右子树
int leftSubtreeSize = rootIndex - inStart;
root.left = buildTreeHelper(preorder, preStart + 1, preStart + leftSubtreeSize, inorder, inStart, rootIndex - 1);
root.right = buildTreeHelper(preorder, preStart + leftSubtreeSize + 1, preEnd, inorder, rootIndex + 1, inEnd);
return root;
}
}
```
其中,buildTree()方法接受先序遍历和中序遍历的数组作为参数,并返回构建好的二叉树的根节点。buildTreeHelper()方法是递归构建二叉树的辅助函数,它接受先序遍历和中序遍历的数组,以及当前子树的范围作为参数,并返回构建好的子树的根节点。在buildTreeHelper()方法中,首先判断递归终止条件,即先序遍历数组或中序遍历数组为空。然后,找到先序遍历数组的第一个节点作为根节点,并在中序遍历数组中找到根节点的位置。接着,递归构建左右子树,并将它们分别作为当前节点的左右子节点。最后,返回当前节点。
阅读全文