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;
    }
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章