利用JAVA根据二叉树的前序和中序输出一个二叉树
时间: 2024-09-06 14:02:19 浏览: 32
在Java中,利用前序遍历(根-左-右)和中序遍历(左-根-右)重建一棵二叉树是一项常见的算法挑战。这是因为前序和中序遍历可以唯一确定一棵二叉树的结构。下面是一个简单的步骤来解决这个问题:
1. **创建节点类**:首先定义一个二叉树节点的类,包含值、左右子节点。
```java
class TreeNode {
int val;
TreeNode left;
TreeNode right;
// constructor, getters and setters...
}
```
2. **递归函数**:创建两个递归方法,`buildTreeHelper`用于实际的构建过程。
- `preorder`是前序遍历的结果数组,`inorder`是中序遍历的结果数组。
- `index`表示当前处理到的中序遍历的位置。
```java
private TreeNode buildTreeHelper(int[] preorder, int[] inorder, int preStart, int inEnd) {
if (preStart == inEnd) return null; // 如果遍历结束,返回null
TreeNode root = new TreeNode(preorder[preStart]); // 根节点
int i = index; // 中序遍历查找根节点的位置
while (i < inEnd && inorder[i] != preorder[preStart]) { // 找到根节点在中序中的位置
i++; // 继续向前寻找
}
root.left = buildTreeHelper(preorder, inorder, preStart + 1, i); // 构建左子树
root.right = buildTreeHelper(preorder, inorder, i + 1, inEnd); // 构建右子树
return root;
}
```
3. **调用函数**:最后,在主程序中,传入前序和中序遍历的结果,以及开始和结束索引,构建整棵树。
```java
public static TreeNode buildTree(int[] preorder, int[] inorder) {
return buildTreeHelper(preorder, inorder, 0, preorder.length - 1);
}
```
阅读全文