当前位置:   article > 正文

力扣-106 从中序与后序遍历序列构造二叉树_从中序与后序遍历序列构造二叉树(力扣106)

从中序与后序遍历序列构造二叉树(力扣106)

给定两个整数数组inorder和postorder,前者是二叉树的中序遍历,后者是二叉树的后序遍历,请构造并返回这棵二叉树

首先要理解二叉树的中序遍历和后序遍历的特点,后序数组的最后一个元素一定是当前子树的根节点,由于中序遍历是左右中,因此利用后序数组的最后一个元素,然后在中序数组中进行切割就可以得到当前子树的左前序数组右前序数组,由于后序遍历是左右中,因此可以利用左前序数组和右前序数组元素个数在后序数组中切割得到左后序数组右后序数组,然后利用这些数组递归构造当前子树的左右子树即可

代码如下:

  1. TreeNode* traversal(vector<int>& inorder, vector<int>& postorder){
  2. if(postorder.size() == 0) return nullptr;
  3. //确定根节点的值,是后序数组的最后一个元素
  4. int rootVal = postorder[postorder.size()-1];
  5. TreeNode* root = new TreeNode(rootVal);
  6. //节点是叶子节点
  7. if(postorder.size() == 1) return root;
  8. //确定中序数组切割点
  9. int index=0;
  10. for(;index<inorder.size();index++){
  11. if(inorder[index] == rootVal)
  12. break;
  13. }
  14. //切中序数组,得到左中序数组和右中序数组
  15. //数组统一为左闭右开区间
  16. vector<int> leftInorder(inorder.begin(),inorder.begin() + index);
  17. vector<int> rightInorder(inorder.begin() + index + 1,inorder.end());
  18. //后序数组需要舍弃末尾元素,以便确定下一次递归的根节点
  19. postorder.resize(postorder.size()-1);
  20. //切割后序数组,得到左后序和右后序数组
  21. //数组统一为左闭右开区间
  22. vector<int> leftPostorder(postorder.begin(),postorder.begin() + leftInorder.size());
  23. vector<int> rightPostorder(postorder.begin() + leftInorder.size(),postorder.end());
  24. //递归处理
  25. root->left = traversal(leftInorder,leftPostorder);
  26. root->right = traversal(rightInorder,rightPostorder);
  27. return root;
  28. }

本题和力扣105 从前序与中序遍历序列构造二叉树本质上是一样的,前序数组的第一个元素是当前子树的根节点,与后序数组类似。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Cpp五条/article/detail/465164
推荐阅读
相关标签
  

闽ICP备14008679号