Given inorder and postorder traversal of a tree, construct the binary tree.
Note:
You may assume that duplicates do not exist in the tree.
题目解析:
用中序遍历和后续遍历还原二叉树。并且二叉树的节点的值没有重复的。这样就好做了,我们知道后序遍历最后的节点是根节点,所以直接可以确定根节点,再在中序遍历的数组中找值为根节点的下标index,这个下标之前的所有值都是左子树的节点值得集合,求出集合的大小之后len,在后序遍历中有数组开始的len个节点都是左子树的。由此我们可以得到左子树的中序遍历以及后续遍历的数组;同理,我们也可知道右子树的中序遍历以及后续遍历的数组,这样我们便可以用递归来实现。代码如下:
/**
* 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* BT(vector<int>&inorder,int instart,int inend,vector<int>&postorder,int pstart,int pend)
{
if(instart>inend) return NULL;
TreeNode* root=new TreeNode(postorder[pend]);
int index;
for(int i=instart;i<=inend;i++)
{
if(inorder[i]==root->val)
{
index=i;
break;
}
}
int len=index-instart;
TreeNode* left=BT(inorder,instart,index-1,postorder,pstart,pstart+len-1);
TreeNode* right=BT(inorder,index+1,inend,postorder,pstart+len,pend-1);
root->left=left;
root->right=right;
return root;
}
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
if(inorder.size()==0) return NULL;
if(inorder.size()!=postorder.size()) return NULL;
TreeNode* node=BT(inorder,0,inorder.size()-1,postorder,0,postorder.size()-1);
return node;
}
};