2020.5.4 从前序与中序遍历序列构造二叉树 中等 105

Posted 2021-02-20 wasi-991017

解题思路

1. 看到题目想了想就知道用递归了，不断划分左右子树；
2. 这道题做了很久，因为对指针引用的不熟悉，一开始我使用的递归函数的返回类型为void，是通过指针作为递归函数的参数，不断更新树，最后改成这种容易看的方式;
3. 主要是要注意划分左右子树的索引，改了很多次都是因为递归时传右子树索引总是传错。

代码

/**
 * 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* recursion(vector<int> pre,vector<int> vin, int preBegin, int vinBegin, int vinEnd){
        if(preBegin>=pre.size() || vinBegin>=vinEnd) return NULL;
        TreeNode* res=new TreeNode(pre[preBegin]);
        for(int i=vinBegin; i<vinEnd; i++){
            if(vin[i]==pre[preBegin]){
                res->left=recursion(pre, vin, preBegin+1, vinBegin, i);
                res->right=recursion(pre, vin, preBegin+(i-vinBegin)+1, i+1, vinEnd);
                break;
            } 
        }
        return res;
    }

    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if(preorder.size()==0 || inorder.size()==0 || preorder.size()!=inorder.size()) return NULL;
        
        return recursion(preorder, inorder, 0, 0, inorder.size());
    }
};