leetcode105

Posted 2021-01-10 AsenYang

/**
 * 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* buildTree(vector<int>& preorder, vector<int>& inorder) {
        if (preorder.empty())
            return NULL;
        int root_index = 0;
        return createBST(preorder,inorder,root_index,preorder.size()-1,root_index);
    }
    TreeNode* createBST(vector<int>& preorder, vector<int>& inorder, int start, int end, int& index) {
        int v = preorder[index];
        int i = start;
        for (i; i <= end; i++)
            if (v == inorder[i])
                break;
        TreeNode* root = new TreeNode(v);
        if (i-1 >= start)
            root->left = createBST(preorder,inorder,start,i-1,++index);
        if (end >= i+1)
            root->right = createBST(preorder,inorder,i+1,end,++index);
        return root;
    }
};

这是一道分治的题目，用先序找到根节点，用中序找到其左右子树。