二叉树进阶——迭代实现前中后序遍历

╰+哭是因爲堅強的太久メ 2022-10-07 04:53 185阅读 0赞

二叉树的递归遍历代码简单且容易理解,具体可以看这篇博客,里面的递归解释的较为详细
数据结构——二叉树的链式结构及实现(C语言)

然而现实生活中,一棵树的节点往往较多且深度更深时,如果继续使用递归就会产生较多的栈帧,导致栈溢出,对于Linux的进程地址空间来说,一个进程所分配的栈的空间大小是有限且较小的,因此这时候我们就需要使用迭代的方式来实现一棵树的遍历。

不过对于现代编译器而言,debug版本下的递归较大时会导致栈溢出,但是release版本编译器做了优化往往和迭代的效率差不多

尽管这样,迭代方式还是要掌握的,尤其后序遍历是大厂面试的高频考点!

接下来由浅入深详细理解二叉树的迭代遍历!

一、前序遍历

1.1 问题描述

144. 二叉树的前序遍历

给你二叉树的根节点 root ,返回它节点值的 前序 遍历。

示例 1:

img

  1. 输入:root = [1,null,2,3]
  2. 输出:[1,2,3]

示例 2:

  1. 输入:root = []
  2. 输出:[]

示例 3:

  1. 输入:root = [1]
  2. 输出:[1]

示例 4:

img

  1. 输入:root = [1,2]
  2. 输出:[1,2]

示例 5:

img

  1. 输入:root = [1,null,2]
  2. 输出:[1,2]

提示:

  • 树中节点数目在范围 [0, 100]
  • `-100 <= Node.val <= 100

1.2 解题思路

首先回顾一下前序遍历的顺序是解题关键

  • 先访问根节点
  • 遍历访问左子树
  • 遍历访问右子树

那么迭代实现二叉树三种遍历的话,都需要借助一个栈来保存树的节点,以便于回溯访问其它节点,定义一个cur指针指向当前节点

注意

栈中保存的是节点的指针而非val值,画图是为了方便理解不要产生误解

前序遍历迭代实现步骤:

  1. 首先遍历树的左子树节点并依次保存到栈st中和数组中直到null停止,如下图所示,遍历并输出的左子树的顺序就是前序遍历的顺序
    在这里插入图片描述
    当这一步骤完成后我们会发现树的左子树全部访问完毕只剩右子树没有访问了,这个时候我们区访问当前节点cur的右子树
  2. 左子树为null,从栈顶取出节点作为当前节点,访问当前节点右子树,并更新cur,重复步骤1
    在这里插入图片描述
    在这里插入图片描述

简单来说,就是不断访问左子树入栈直到null,然后取栈顶元素出栈访问左子树为null的节点的右子树,将这个右子树的根节点作为当前节点继续访问左子树入栈。

上面叙述的1,2步骤代码实现简单,主要难点在于while主循环的结束条件

  • cur不为空,cur不为空代表当前树还没有访问完,添加这个条件还有一个原因就是循环开始时栈为空,不能单纯以栈作为结束条件
  • st栈不为空,栈不为空代表栈中还有数据没有访问完

具体代码如下,相信结合代码更容易理解

1.3 代码实现

  1. /**
  2. * Definition for a binary tree node.
  3. * struct TreeNode {
  4. * int val;
  5. * TreeNode *left;
  6. * TreeNode *right;
  7. * TreeNode() : val(0), left(nullptr), right(nullptr) {}
  8. * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
  9. * TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
  10. * };
  11. */
  12. class Solution {
  13. public:
  14. vector<int> preorderTraversal(TreeNode* root) {
  15. vector<int> ret; //ret保存遍历结果
  16. stack<TreeNode *> st; //st保存树的节点指针
  17. TreeNode *cur = root; //cur指向当前遍历到的节点
  18. //cur不为空代表树没访问完, st不为空代表栈中还有数据
  19. while(cur || !st.empty())
  20. {
  21. //遍历左子树
  22. while(cur)
  23. {
  24. st.push(cur); //当前节点入栈
  25. ret.push_back(cur->val); //当前节点值保存到数组中
  26. cur = cur->left; //继续遍历左子树
  27. }
  28. //左子树遍历完,从栈顶取出top并指向右子树继续遍历
  29. TreeNode *top = st.top();
  30. st.pop();
  31. cur = top->right;
  32. }
  33. return ret;
  34. }
  35. };

如果理解了前序遍历,中序遍历就更容易实现了,只是简单访问位置的变化

二、中序遍历

1.1 问题描述

94. 二叉树的中序遍历

给定一个二叉树的根节点 root ,返回它的 中序 遍历。

示例 1:

img

  1. 输入:root = [1,null,2,3]
  2. 输出:[1,3,2]

示例 2:

  1. 输入:root = []
  2. 输出:[]

示例 3:

  1. 输入:root = [1]
  2. 输出:[1]

示例 4:

img

  1. 输入:root = [1,2]
  2. 输出:[2,1]

示例 5:

img

  1. 输入:root = [1,null,2]
  2. 输出:[1,2]

提示:

  • 树中节点数目在范围 [0, 100]
  • -100 <= Node.val <= 100

1.2 解题思路

中序遍历的解题思路和前序遍历相同,前序遍历是遍历左子树并入栈输出,中序遍历先遍历左子树并入栈直到null但不输出,当左子树访问完从栈顶取节点访问右子树时再打印即可,这正好符合了中序遍历的顺序

1.3 代码实现

  1. /**
  2. * Definition for a binary tree node.
  3. * struct TreeNode {
  4. * int val;
  5. * TreeNode *left;
  6. * TreeNode *right;
  7. * TreeNode() : val(0), left(nullptr), right(nullptr) {}
  8. * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
  9. * TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
  10. * };
  11. */
  12. class Solution {
  13. public:
  14. vector<int> inorderTraversal(TreeNode* root) {
  15. vector<int> ret; //ret保存遍历结果
  16. stack<TreeNode *> st; //st保存树的节点指针
  17. TreeNode *cur = root; //cur指向当前遍历到的节点
  18. //cur不为空代表树没访问完, st不为空代表栈中还有数据
  19. while(cur || !st.empty())
  20. {
  21. //遍历左子树
  22. while(cur)
  23. {
  24. st.push(cur); //当前节点入栈
  25. cur = cur->left; //继续遍历左子树
  26. }
  27. //左子树遍历完,从栈顶取出cur并指向右子树继续遍历
  28. cur = st.top();
  29. st.pop();
  30. ret.push_back(cur->val); //当前节点值保存到数组中
  31. cur = cur->right;
  32. }
  33. return ret;
  34. }
  35. };

后序遍历

3.1 问题描述

145. 二叉树的后序遍历

给定一个二叉树,返回它的 后序 遍历。

示例:

  1. 输入: [1,null,2,3]
  2. 1
  3. \
  4. 2
  5. /
  6. 3
  7. 输出: [3,2,1]

3.2 解题思路

后续遍历的顺序为

  • 遍历访问左子树
  • 遍历访问右子树
  • 访问根节点

上面两个遍历的思路是遇到左子树为null压栈再访问右子树,无非就是访问的顺序不同

因为左节点只要不为null就会一直压栈,而后序遍历输出当前节点的标志就是左右子树都为null,这种情况表明节点左右子树为空已经遍历完了,访问根节点,符合后序遍历顺序

在这里插入图片描述
在这里插入图片描述

3.3 代码实现

  1. /**
  2. * Definition for a binary tree node.
  3. * struct TreeNode {
  4. * int val;
  5. * TreeNode *left;
  6. * TreeNode *right;
  7. * TreeNode() : val(0), left(nullptr), right(nullptr) {}
  8. * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
  9. * TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
  10. * };
  11. */
  12. class Solution {
  13. public:
  14. vector<int> postorderTraversal(TreeNode* root) {
  15. vector<int> ret; //ret保存遍历结果
  16. stack<TreeNode *> st; //st保存树的节点指针
  17. TreeNode *prev = nullptr;
  18. TreeNode *cur = root; //cur指向当前遍历到的节点
  19. //cur不为空代表树没访问完, st不为空代表栈中还有数据
  20. while(cur || !st.empty())
  21. {
  22. //遍历左子树
  23. while(cur)
  24. {
  25. st.push(cur); //当前节点入栈
  26. cur = cur->left; //继续遍历左子树
  27. }
  28. //左子树遍历完,从栈顶取出cur并指向右子树继续遍历
  29. TreeNode *top = st.top();
  30. //右子树为null代表当前节点top左右子树都为空则输出,
  31. //但是可能死循环,因为判断节点右子树时可能会被访问两次
  32. //访问两次的标志就是当前节点的右树等于上一次访问的节点
  33. if(top->right == nullptr || top->right == prev)
  34. {
  35. ret.push_back(top->val);
  36. st.pop();
  37. prev = top;
  38. cur = nullptr;
  39. }
  40. else
  41. {
  42. cur = top->right;
  43. }
  44. }
  45. return ret;
  46. }
  47. };

发表评论

表情:
评论列表 (有 0 条评论,185人围观)

还没有评论,来说两句吧...

相关阅读