题目:
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
题解:
已知中序,后序,即可唯一确定一棵二叉树,关键点抓住后序的第一个元素是当前根节点,因而我们可以将序列进行切割,递归到只有一个元素时,构造节点。
具体解释贴在代码里
/**
* 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>& inorder, vector<int>& postorder,int in_st,int in_ed,int po_st,int po_ed) {
//如果没有元素说明是空节点,要置空返回,不可少,因为预先没有检查范围
if(in_st>in_ed) return NULL;
TreeNode* root=new TreeNode(postorder[po_ed]);
if(in_st==in_ed)
return root;
int loc=-1;
//在中序遍历中找到当前根节点
for(int i=in_st;i<=in_ed;i++)
{
if(inorder[i]==postorder[po_ed])
{
loc=i;
break;
}
}
//这里一共4个边界点,我使用闭区间,左右点都能取到,这里是易错点
//我使用的是边界点,再根据区间长度确定另一个边界点的做法
//root->left指向左区间,root->right指向右区间
root->left = buildTree(inorder, postorder, in_st, loc-1, po_st, po_ed-in_ed+loc-1);
root->right = buildTree(inorder, postorder, loc+1, in_ed, po_ed-in_ed+loc, po_ed-1);
//最后记得把root返回上去
return root;
}
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
return buildTree(inorder,postorder,0,inorder.size()-1,0,postorder.size()-1);
}
};