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 34 35 36
| class Solution { public: TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) { if(preorder.size() != inorder.size()) return NULL; int length = preorder.size(); return constructTree(preorder,0,length-1,inorder,0,length-1); } TreeNode* constructTree(vector<int>& preorder,int ps, int pend,vector<int>& inorder ,int is,int iend){ if(ps>pend || is>iend) return NULL; TreeNode *root = new TreeNode(preorder[ps]); int irootpos; for(int i=is;i<=iend;i++){ if(inorder[i] == root->val){ irootpos = i; break; } } int leftlength = irootpos - is; int rightlength = iend - irootpos; root->left = constructTree(preorder,ps+1,ps+leftlength,inorder,is,irootpos-1); root->right = constructTree(preorder,pend-rightlength+1,pend,inorder,irootpos+1,iend); return root; } };
|