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


?做题思路or感想:

  1. 根据前序遍历找根节点
  2. 由前序遍历根节点的值来切割中序数组,再由中序数组切割前序数组,最终切割成左子树的中序,前序数组和右子树的中序,前序数组。
  3. 递归造根节点的左子树和右子树
class Solution {
public:
    TreeNode* buildTree(vector& preorder, vector& inorder) {
        if (preorder.size() == 0 || inorder.size() == 0)return nullptr;	//如果数组空,则是空节点
        int val = preorder[0];	//找前序数组的第一个,便是根节点的值
        TreeNode* root = new TreeNode(val);
        if (preorder.size() == 1)return root;	//找到的是最后一个根节点,则返回
        int index;
        //切割中序数组
        for (int i = 0; i < inorder.size(); i++) {
            if (inorder[i] == val) {
                index = i;
                break;
            }
        }
        //造左子树,右子树的前,中序数组
        vectorleftPre (preorder.begin() + 1, preorder.begin()+ 1 + index);
        vectorleftIn (inorder.begin(), inorder.begin() + index);
        vectorrightPre (preorder.begin() + index + 1, preorder.end());
        vectorrightIn (inorder.begin() + index + 1, inorder.end());
       	//造出左子树,右子树
        root->left = buildTree(leftPre, leftIn);
        root->right = buildTree(rightPre, rightIn);
        return root;
    }
};