leetcode105—— 从前序与中序遍历序列构造二叉树

/*
 * @lc app=leetcode.cn id=105 lang=cpp
 *
 * [105] 从前序与中序遍历序列构造二叉树
 */

// @lc code=start
/**
 * 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:
    /*
    1）前序数组的第一个元素为根节点，根据根节点可将中序数组划分为左子树中序列表和右子树中序数组
    2）又因为一棵树的中序数组长度与前序数组长度大小相同，所以可以根据长度划分前序数组为左右子树前序数组
    3）对于根的左右结点，递归调用即可
    */
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
       if(preorder.empty()||inorder.empty()) return nullptr;

       TreeNode *root=new TreeNode(*preorder.begin());//begin返回的是迭代器，要取值的话加 *

       auto it=find(inorder.begin(),inorder.end(),*preorder.begin());

       vector<int> inleft(inorder.begin(),it); //左闭右开，所以inleft不包含it这个值
       vector<int> inright(it+1,inorder.end());//加 1是为了跳过根节点
       int sz=inleft.size();
       vector<int> preleft(preorder.begin()+1,preorder.begin()+1+sz);
       vector<int> preright(preorder.begin()+1+sz,preorder.end());
       root->left=buildTree(preleft,inleft);
       root->right=buildTree(preright,inright);

       return root;

    }
};
// @lc code=end