leetcode刷题题解——889. 根据前序和后序遍历构造二叉树

最新推荐文章于 2024-09-16 06:54:08 发布

绿豆蛙给生活加点甜

最新推荐文章于 2024-09-16 06:54:08 发布

阅读量4.1k

点赞数

分类专栏： LeetCode刷题汇总文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/qq_53226437/article/details/122901973

版权

LeetCode刷题汇总专栏收录该内容

42 篇文章 0 订阅

订阅专栏

该博客介绍了如何通过给定的前序遍历和后序遍历数组来构建二叉树。核心思路是利用前序遍历的第一个元素作为根节点，然后寻找后序遍历中左子树的末尾节点来确定左子树的长度，进而递归地构建左右子树。此算法解决了二叉树序列化与反序列化的问题。

摘要由CSDN通过智能技术生成

private int[] pre
private int[] post;

public TreeNode constructFromPrePost(int[] preorder, int[] postorder) {
    pre = preorder;
    post = postorder;
    return buildTree(0,0,pre.length);
}

private TreeNode buildTree(int i,int j,int n){
    if (n==0) return null;
    TreeNode root = new TreeNode(pre[i++]);
    if (n==1) return root;
    int L = 1;
    for (;L<n;L++) if (post[j+L-1]==pre[i])  break; 
    root.left = buildTree(i,j,L);
    root.right = buildTree(i+L,j+L,n-1-L);

    return root;
}

思路：
首先要知道
前序遍历
[根节点，(左子树),(右子树)]
后序遍历
[(左子树)，(右子树)，根节点]
那么，我们可以知道根节点就是前序遍历中的首元素，那么怎么将左右子树划分开来呢？如果我们知道左子树的长度是不是就可以划分开左右子树了呢？那长度又怎么获取呢？
前序遍历中左子树的根节点会出现在后序遍历的左子树的哪里呢？
答案是：最后一个
即有：pre[1]==post[L-1]
设i为前序遍历左子树的首位元素索引，j为后序遍历左子树首位元素的索引，有pre[i]=post[j+L-1]
由此即可划分左右子树进行接下来的递归

将preorder和postorder存储到类中的成员变量，减少了传参时的繁琐
初始时，前序遍历左子树的首位元素索引是1，但是答案里传的是0，这是因为需要先创建根节点，创建之后i就会自增1，而后序遍历的左子树首位元素索引是0，也就是传了0，并且传了子树节点的长度
如果该子树长度为0，直接返回null，为1，则创建根节点后返回，长度大于1，继续划分左右子树
循环获取左子树长度后，继续递归创建左右子树
返回root