(2022.02.16)每日一题 通过中序和后序遍历构造二叉树
这几天家里有些事情,学校里有些事情需要去处理,同时去面了个试找了个实习,做三维重建和视觉相关的工作。而且也有些烦心和焦虑的事情,所以耽搁了许久,之后还是得要恢复学习。
今日做题依旧是二叉树遍历相关的问题,思路是有,但还是写了蛮久的,但是我总结一下,中序遍历的索引,你只需要去考虑中序的特征,后序遍历,考虑后序的特征,放在一起考虑,反而脑子转不过来,这么简单的道理,我也不知道为啥我才明白,分别考虑index的特征,去计算即可。
class Solution {
private:
unordered_map<int,int> num2index;
public:
TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
TreeNode* ans = new TreeNode();
int size = inorder.size();
for(int i =0;i<size;i++){
num2index[inorder[i]] = i;
}
ans = recoverTree(inorder,postorder,0,size-1,0,size-1);
return ans;
}
TreeNode* recoverTree(vector<int>& inorder, vector<int>& postorder,int left_in,int right_in,int left_post,int right_post){
if(left_in>right_in){
return nullptr;
}
TreeNode* root = new TreeNode(postorder[right_post]);
int left_size = num2index[postorder[right_post]] - left_in;
root->left = recoverTree(inorder,postorder,left_in,num2index[postorder[right_post]]-1,left_post,left_post+left_size-1);
root->right = recoverTree(inorder,postorder,num2index[postorder[right_post]]+1,right_in,left_post+left_size,right_post-1);
return root;
}
};