Given preorder and inorder traversal of a tree, construct the binary tree.
Note:
You may assume that duplicates do not exist in the
tree.
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33 |
/** * Definition for binary tree * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */class
Solution {public: TreeNode *DFS(vector<int> &preorder, vector<int> &inorder,int
leftstart,int
leftend,int
rightstart,int
rightend) { if(leftend-leftstart<0||rightend-rightstart<0) return
NULL; TreeNode *root=new
TreeNode(preorder[leftstart]); int
pos;//中序遍历中根结点位置 for(int
i=rightstart;i<=rightend;i++) { if(inorder[i]==preorder[leftstart]) { pos=i; break; } } int
len=pos-rightstart; root->left=DFS(preorder,inorder,leftstart+1,leftstart+len,rightstart,pos-1); root->right=DFS(preorder,inorder,leftstart+len+1,leftend,pos+1,rightend); return
root; } TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) { return
DFS(preorder,inorder,0,preorder.size()-1,0,inorder.size()-1); }}; |
[LeetCode]Construct Binary Tree from Preorder and Inorder Traversal,布布扣,bubuko.com
[LeetCode]Construct Binary Tree from Preorder and Inorder Traversal
原文:http://www.cnblogs.com/Rosanna/p/3594708.html