题目描述:
根据中序遍历和后序遍历树构造二叉树
注意事项:
你可以假设树中不存在相同数值的节点
样例:
给出中序遍历:[1,2,3]和后序遍历:[1,3,2]. 返回如下的树:
思路跟上一篇一样,唯一的不同是这里是从后序遍历找出根节点,即后序遍历数组的最后一个是根节点。上一篇是从先序遍历找出根节点,即先序遍历数组的第一个是根节点。
代码详解:
class Solution {
public:
/*
* @param inorder: A list of integers that inorder traversal of a tree
* @param postorder: A list of integers that postorder traversal of a tree
* @return: Root of a tree
*/
TreeNode * buildTree(vector<int> &inorder, vector<int> &postorder) {
// write your code here
TreeNode *root;
if(inorder.size()==0)
{
root=NULL;
return root;
}
int rootNode=postorder[postorder.size()-1];
int index=findpos(inorder,rootNode);
vector<int>left_inorder;
vector<int>right_inorder;
for(int i=0;i<index;i++){
left_inorder.push_back(inorder[i]);
}
for(int i=index+1;i<inorder.size();i++){
right_inorder.push_back(inorder[i]);
}
vector<int>left_postorder;
vector<int>right_postorder;
for(int i=0;i<index;i++)
{
left_postorder.push_back(postorder[i]);
}
for(int i=index;i<postorder.size()-1;i++)
{
right_postorder.push_back(postorder[i]);
}
root=new TreeNode();
if(root!=NULL)
{
root->val=rootNode;
root->left=buildTree(left_inorder,left_postorder);
root->right=buildTree(right_inorder,right_postorder);
}
return root;
}
int findpos(vector<int>inorder,int m)
{
for(int i=0;i<inorder.size();i++)
{
if(inorder[i]==m)
{
return i;
}
}
return 0;
}
};