大家好,我是三叔,很高兴这期又和大家见面了,一个奋斗在互联网的打工人。
笔者在从中序与后序遍历序列构造二叉树怎么写一文中有介绍如何通过中序遍历和后续遍历构造二叉树,难度还是挺高的,重点在于找到中间节点,并且左右分界要弄清楚,这篇文章我写出了前序遍历和中序遍历如何构造二叉树的Java编程代码,具体步骤我都在代码中用注解表示出来,一句话,一行代码,这样就不会写错。
上代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
// 力扣#105题
class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
if(preorder.length == 0 || inorder.length == 0) {
return null;
}
return buildTreeNode(preorder, 0 , preorder.length,
inorder, 0, inorder.length);
}
private TreeNode buildTreeNode(int[] preorder, int prevStart, int prevLen,
int[] inorder, int inorderStart, int inorderLen) {
// if(prevStart == prevLen) {
// return null;
// }
if (prevStart >= prevLen || inorderStart >= inorderLen) { // 不满足左闭右开,说明没有元素,返回空树
return null;
}
// 开闭区间:左闭右开 ---》 [left, right)
// 中间节点 前序遍历的第一个
int rootVal = preorder[prevStart]; // 前序遍历的第一个开始节点就是中间节点(根节点)
TreeNode root = new TreeNode(rootVal);// 根节点放进去(构造中间节点)
// 现在找到中序遍历的中间节点,进行分割
int midIndex;
for(midIndex = 0; midIndex < inorderLen; midIndex++) {
if(inorder[midIndex] == rootVal) { // 在中序遍历中找到中间节点进行分割,
break;
}
}
// 此时的midIndex 就是中序数组的中间节点的位置
// 此时有一个很重的点,就是中序数组大小一定是和后序数组的大小相同的(这是必然)
// 中序数组我们都切成了左中序数组和右中序数组了,那么后序数组就可以按照左中序数组的大小来切割,切成左后序数组和右后序数组
int lenOfLeft = midIndex - inorderStart; // 保存中序左子树个数,用来确定前序数列的左前序个数
// 子树左:前序遍历的的左数组, 中序遍历的左数组,
int preleftStart = prevStart + 1;
int preLeftEnd = prevStart + lenOfLeft + 1;
int midLeftStart = inorderStart;
int midLeftEnd = midIndex;
// 子树右:前序遍历的右数组,中序右数组
int preRightStart = prevStart + lenOfLeft + 1;
int preRightEnd = prevLen;
int midRightStart = midIndex + 1;
int midRightEnd = inorderLen;
root.left = buildTreeNode(preorder, preleftStart, preLeftEnd, inorder, midLeftStart, midLeftEnd);
root.right = buildTreeNode(preorder, preRightStart, preRightEnd, inorder, midRightStart, midRightEnd);
return root;
}
}
思考
前中序遍历和中后续遍历都可以写出二叉树,那前序遍历和后序遍历可以写出二叉树吗?
答案是:不可以。
因为前序和后序不能唯一确定一棵二叉树!,因为没有中序遍历无法确定左右部分,也就是无法分割。