有沒有深度學習大佬,圖像處理方向的,帶帶我
問題描述:
根據一棵樹的前序遍歷與中序遍歷構造二叉樹。
來源:力扣(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;
}
};