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

Posted 2020-12-18 pesuedream

 

 

 解答:本题主要采用递归的方式来构造二叉树

解答思路：

1、前序遍历的第一个节点必然等于后序遍历的最后一个节点，这个节点为根节点，然后再找一把刀，把中间的砍成左子树和右子树

2、前序的第二个节点，就是左子树的根节点，同时要找到这个节点在后序中的位置，那么从后序开头到这个位置之间，就是左子树了，这个位置到后序的末尾就是右子树了。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* constructFromPrePost(vector<int>& pre, vector<int>& post) {
        int length=pre.size()-1;
        TreeNode* result;
        result=helper(pre,post,0,length,0,length);
        return result;

    }
    TreeNode* helper(vector<int>& pre,vector<int>& post,int preStart,int preEnd,int postStart,int postEnd){
        if(preStart>preEnd) return NULL;
        if(preStart==preEnd){
            return new TreeNode(pre[preStart]);
        }
        TreeNode* root=new TreeNode(pre[preStart]);
        int val=pre[preStart+1];
        for(int i=postStart;i<=postEnd;i++){
            if(post[i]==val){
                int length=i-postStart;
                root->left=helper(pre,post,preStart+1,preStart+length+1,postStart,i);
                root->right=helper(pre,post,preStart+length+2,preEnd,i+1,postEnd-1);
                
            }
        }
        return root;
    }
};

作者：a380922457
链接：https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-postorder-traversal/solution/jian-dan-yi-dong-ban-by-a380922457-3/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。