Hello吖,各位小伙伴好久不见,十分想念,已经有一段时间没有更新博客了,具体原因嘛(放在文末)
题目: 剑指 Offer 07. 重建二叉树 ,我们今天来看一道递归的题目吧,这是选自剑指Offer上的一道题,好了,我们一起来看看题意吧:
考虑到直接复制题目,或者截屏的方式不是很方便阅读,我就把直接题目链接放下面!
题目传送门: 剑指 Offer 07. 重建二叉树
思路
:
可以先看看代码(有注释的),然后再来看看这个思路
本题的一个难点就是:找到左右子树在前序和中序遍历序列中对应的区间
在前序序列中取得根节点(序列第一个),然后在中序序列中找到根节点的位置记为pos,
中序序列pos左边的则为左子树,对应数量为 pos - 中序序列起始位置(inStart) ,
将中序序列pos左边的节点数量记录为x,x = pos - inStart
二叉树的左子树在前序序列中的区间为:preStart(前序序列中的第一个位置)+1到preStart+x,在中序序列中的位置为inStart(中序序列中的第一个位置)到pos-1
二叉树的右子树在前序序列中的区间为:preStart+x+1,到preEnd(前序序列的最后一个位置),在中序序列中的位置为pos+1到inEnd(中序序列中的最后一个位置)
递归处理以上步骤
我们来看看成功AC的代码吧:
class Solution { vector<int> pre,in; public: TreeNode* dfs(int preStart,int preEnd,int inStart,int inEnd){ if(preStart>preEnd||inStart>inEnd) return NULL; //前序遍历中的开始节点就是根节点,取出根节点值 int t = pre[preStart]; //寻找子树的根节点在中序序列中的哪个位置 int pos; for(int i=0;i<in.size()-1;i++){ if(in[i]==t){ pos = i; break; }} auto s = new TreeNode(t); //【pos-inStart】为左子树中的节点数 int x = pos - inStart; //递归构建左子树,并连上根节点 //前序遍历中的区间【preStart+1,preStart+x】,中序遍历中的区间【inStart,pos-1】 s->left = dfs(preStart+1,preStart+x,inStart,pos-1); //递归构建右子树,并连上根节点 //前序遍历中的区间【preStart+1+x,preEnd】,中序遍历中的区间【pos+1,inEnd】 s->right = dfs(preStart+1+x,preEnd,pos+1,inEnd); return s; } TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) { pre = preorder; in = inorder; return dfs(0,pre.size()-1,0,in.size()-1); } };
💬
哈哈哈,简单说一下前面说的情况额,我就用一句来说明,读孙子兵法,品启强人生😎懂了吧🤪
今天刚好也是元宵节,祝大家元宵快乐,2023一路狂飙😊
结语