LeetCode每日一题5月22日 LeetCode105.前序中序生成树

有没有深度学习大佬，图像处理方向的，带带我

问题描述：

根据一棵树的前序遍历与中序遍历构造二叉树。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal

注意:
你可以假设树中没有重复的元素。

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

解题思路：

模板题，递归，从中序中找根节点，分为左右子树，递归

/**
 * 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) {
        return pre_order(0,preorder.size()-1,0,inorder.size()-1,preorder,inorder);
    }
    TreeNode* pre_order(int leftpre,int rightpre,int leftin,int rightin,vector<int> &pre,vector<int> &in){
        if(leftpre>rightpre || leftin>rightin) return NULL;
        TreeNode* root = new TreeNode(pre[leftpre]);
        int rootin = leftin;
        while(rootin<=rightin && in[rootin]!=pre[leftpre]) rootin++;
        int leftnum = rootin-leftin;
        root->left = pre_order(leftpre+1,leftpre+leftnum,leftin,rootin-1,pre,in);
        root->right = pre_order(leftpre+leftnum+1,rightpre,rootin+1,rightin,pre,in);
        return root;
    }
};