问题:
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
可以假设树中没有重复的元素。
例如,给出中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
解答思路:
以下是使用 Java 实现根据中序遍历和后序遍历序列构造二叉树的代码:
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int val) {
this.val = val;
}
}
public class BuildBinaryTreeFromInorderAndPostorder {
public TreeNode buildTree(int[] inorder, int[] postorder) {
return buildTreeHelper(inorder, 0, inorder.length - 1, postorder, 0, postorder.length - 1);
}
public TreeNode buildTreeHelper(int[] inorder, int inStart, int inEnd, int[] postorder, int postStart, int postEnd) {
if (inStart > inEnd || postStart > postEnd) {
return null;
}
int rootVal = postorder[postEnd];
TreeNode root = new TreeNode(rootVal);
int rootIndexInInorder = 0;
for (int i = inStart; i <= inEnd; i++) {
if (inorder[i] == rootVal) {
rootIndexInInorder = i;
break;
}
}
int leftSubtreeSize = rootIndexInInorder - inStart;
root.left = buildTreeHelper(inorder, inStart, rootIndexInInorder - 1, postorder, postStart, postStart + leftSubtreeSize - 1);
root.right = buildTreeHelper(inorder, rootIndexInInorder + 1, inEnd, postorder, postStart + leftSubtreeSize, postEnd - 1);
return root;
}
public static void main(String[] args) {
int[] inorder = {9, 3, 15, 20, 7};
int[] postorder = {9, 15, 7, 20, 3};
BuildBinaryTreeFromInorderAndPostorder builder = new BuildBinaryTreeFromInorderAndPostorder();
TreeNode root = builder.buildTree(inorder, postorder);
// 这里可以根据需要进行二叉树的遍历或其他操作
}
}
这段代码的主要思路是:
- 在后序遍历序列中,最后一个元素就是根节点的值。
- 在中序遍历序列中,找到根节点的值,就可以将序列分为左子树的中序遍历序列和右子树的中序遍历序列。
- 根据左子树和右子树的中序遍历序列的长度,可以在后序遍历序列中确定左子树和右子树的后序遍历序列。
- 然后,递归地构建左子树和右子树。
在 'main' 方法中,我们可以对构建好的二叉树进行进一步的操作,比如遍历输出节点值等。
(文章为作者在学习java过程中的一些个人体会总结和借鉴,如有不当、错误的地方,请各位大佬批评指正,定当努力改正,如有侵权请联系作者删帖。)