力扣 889. 根据前序和后序遍历构造二叉树

889. 根据前序和后序遍历构造二叉树

给定两个整数数组，preorder 和 postorder ，其中 preorder 是一个具有 无重复 值的二叉树的前序遍历，postorder 是同一棵树的后序遍历，重构并返回二叉树。

如果存在多个答案，您可以返回其中 任何 一个。

示例 1：

输入：preorder = [1,2,4,5,3,6,7], postorder = [4,5,2,6,7,3,1]
输出：[1,2,3,4,5,6,7]

示例 2:

输入: preorder = [1], postorder = [1]
输出: [1]

提示：

• 1 <= preorder.length <= 30
• 1 <= preorder[i] <= preorder.length
• preorder 中所有值都 不同
• postorder.length == preorder.length
• 1 <= postorder[i] <= postorder.length
• postorder 中所有值都 不同
• 保证 preorder 和 postorder 是同一棵二叉树的前序遍历和后序遍历

题解

先序和中序，中序和后序都可以确定一棵树，但是先序和后序确定的树不唯一，题目要求输出一个即可。

一定要先做前两题105. 从前序与中序遍历序列构造二叉树和106. 从中序与后序遍历序列构造二叉树。

依旧是采用递归思想，根据示例一，[1,2,4,5,3,6,7]，[4,5,2,6,7,3,1]，可以发现当preorder下标为cnt=0时，pre[cnt]=1,对应的post中为post[i=6],

接下来分析，针对节点1，如果pre中右边一个的节点2与post中左边一个的节点3不一样，其实代表2不是1的右子树（否则会一样），那就直接把它分配给左子树。

那么首先针对pre中的节点需要在post中遍历找到其对应的节点，然后判断pre右边一个的节点是属于左子树还是右子树。然后建树，就可以得到一个符合题目要求的答案（注意答案不唯一）。

查看代码

 /**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int cnt=0;//计数遍历到preorder的哪个下标
    int len=0;//节点总数
    int flag[3000];//是否访问过的标志位
    TreeNode * build(TreeNode*& cur,vector<int>& postorder,vector<int>& preorder){
        int val=preorder[cnt++];//获取节点值
        int i;
        for(i=0;i<len;i++)//在postorder中寻找节点
            if(postorder[i]==val)
                break;
        flag[i]=1;//标注此节点被访问
        cur=new TreeNode(val);//新建节点
        if(cnt<len&&i>0&&i<len&&preorder[cnt]!=postorder[i-1]&&flag[i-1]!=1)//如果不相等就分给左子树
            cur->left=build(cur->left,postorder,preorder);
        if(cnt<len&&i>0&&i<len&&preorder[cnt]==postorder[i-1]&&flag[i-1]!=1)//否则右子树
            cur->right=build(cur->right,postorder,preorder);
        return cur;//返回建好左右子树的当前节点
    }
    TreeNode* constructFromPrePost(vector<int>& preorder, vector<int>& postorder) {
        TreeNode * root;
        len=postorder.size();
        memset(flag,0,sizeof(len));//初始化len长度为0
        root=build(root,postorder,preorder);
        return root;
    }
};