题目:Given preorder and inorder traversal of a tree, construct the binary tree.
Note:
You may assume that duplicates do not exist in the tree.
思路:
如果所示的二叉树,前序遍历: 4 1 3 2 7 6 5
后序遍历: 3 1 2 4 6 7 5
前序遍历中开始存放的是根节点,所以第一个就是根节点。然后再中序遍历中寻找那个根节点,找到其索引值,然后我们发现从一开始到这个索引值对应的数值,也就是这个根节点,这一部分全部是左边的节点,有半部分全部是右边的节点。
如此递归即可。
代码:
/** * 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) { int preStart=0,preEnd=preorder.size()-1; int inStart=0,inEnd=inorder.size()-1; return constructHelper(preorder,preStart,preEnd, inorder, inStart,inEnd); } TreeNode* constructHelper(vector<int>&preorder,int preStart,int preEnd, vector<int>&inorder,int inStart,int inEnd){ if(preStart>preEnd||inStart>inEnd) return NULL; int val=preorder[preStart]; TreeNode*p =new TreeNode(val); int k=0; for(int i=0;i<inorder.size();i++){ if(inorder[i]==val){ k=i;break; } } p->left= constructHelper(preorder,preStart+1,preStart+(k-inStart), inorder, inStart,k-1);//k之前 //k-start 考虑这种情况,在找到右子树的左子树的时候 //k在右边,inStart也在右边,此时需要找的是左边的个数, //k-inStart 就解决了这个问题 p->right=constructHelper(preorder,preStart+(k-inStart)+1, preEnd, inorder,k+1, inEnd); return p; } };