根据中序遍历和后序遍历树构造二叉树
注意事项
你可以假设树中不存在相同数值的节点
样例
给出树的中序遍历: [1,2,3] 和后序遍历: [1,3,2]
返回如下的树:
2
/ \
1 3
思路:与中序遍历和前序遍历构造二叉树的过程类似。只不过对于后序遍历来说。根节点是最后一个被访问的节点;
或者,可以先将先序遍历序列逆序,然后过程就与中序遍历和前序遍历构造二叉树的过程完全一样了;
注意:最后递归的时候,一定不要把中序遍历的序列和后续遍历的序列弄反了。血的教训,不要直接复制代码,顺着思路过一遍.
/**
* Definition of TreeNode:
* class TreeNode {
* public:
* int val;
* TreeNode *left, *right;
* TreeNode(int val) {
* this->val = val;
* this->left = this->right = NULL;
* }
* }
*/
class Solution {
/**
*@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
*/
/*
思路:与中序遍历和前序遍历构造二叉树的过程类似。只不过对于后序遍历来说。根节点是最后一个被访问的节点;
或者,可以先将先序遍历序列逆序,然后过程就与中序遍历和前序遍历构造二叉树的过程完全一样了;
最后递归的时候,一定不要把中序遍历的序列和后续遍历的序列弄反了。血的教训,不要直接复制代码,顺着思路过一遍
*/
public:
TreeNode *buildTree(vector<int> &inorder, vector<int> &postorder) {
// write your code here
if(postorder.size()==0||inorder.size()==0){
return NULL;
}
if(postorder.size()!=inorder.size()){
return NULL;
}
vector<int> inorder_l,inorder_r,postorder_l,postorder_r;
int root_index=-1;
int len = postorder.size();
TreeNode* root=new TreeNode(postorder[len-1]);
// 在中序队列中找出根节点位置
for(int i=0;i<len;i++){
if(postorder[len-1]==inorder[i]){
root_index=i;
break;
}
}
// 左右子树的后序、中序队列
for(int i=0; i<root_index; i++) {
postorder_l.push_back(postorder[i]);
inorder_l.push_back(inorder[i]);
}
for(int i=root_index+1; i<inorder.size(); i++) {
postorder_r.push_back(postorder[i-1]);
inorder_r.push_back(inorder[i]);
}
root->left=buildTree(inorder_l, postorder_l);
root->right=buildTree(inorder_r, postorder_r);
return root;
}
};