题目描述(中等难度)

94. Binary Tree Inorder Traversal - 图1

二叉树的中序遍历。

解法一 递归

学二叉树的时候,必学的算法。用递归写简洁明了,就不多说了。

  1. public List<Integer> inorderTraversal(TreeNode root) {
  2. List<Integer> ans = new ArrayList<>();
  3. getAns(root, ans);
  4. return ans;
  5. }
  6. private void getAns(TreeNode node, List<Integer> ans) {
  7. if (node == null) {
  8. return;
  9. }
  10. getAns(node.left, ans);
  11. ans.add(node.val);
  12. getAns(node.right, ans);
  13. }

时间复杂度:O(n),遍历每个节点。

空间复杂度:O(h),压栈消耗,h 是二叉树的高度。

官方解法中还提供了两种解法,这里总结下。

解法二 栈

利用栈,去模拟递归。递归压栈的过程,就是保存现场,就是保存当前的变量,而解法一中当前有用的变量就是 node,所以我们用栈把每次的 node 保存起来即可。

模拟下递归的过程,只考虑 node 的压栈。

  1. //当前节点为空,出栈
  2. if (node == null) {
  3. return;
  4. }
  5. //当前节点不为空
  6. getAns(node.left, ans); //压栈
  7. ans.add(node.val); //出栈后添加
  8. getAns(node.right, ans); //压栈
  9. //左右子树遍历完,出栈

看一个具体的例子,想象一下吧。

  1. 1
  2. / \
  3. 2 3
  4. / \ /
  5. 4 5 6
  6. push push push pop pop push pop pop
  7. | | | | |_4_| | | | | | | | | | |
  8. | | |_2_| |_2_| |_2_| | | |_5_| | | | |
  9. |_1_| |_1_| |_1_| |_1_| |_1_| |_1_| |_1_| | |
  10. ans add 4 add 2 add 5 add 1
  11. [] [4] [4 2] [4 2 5] [4 2 5 1]
  12. push push pop pop
  13. | | | | | | | |
  14. | | |_6_| | | | |
  15. |_3_| |_3_| |_3_| | |
  16. add 6 add 3
  17. [4 2 5 1 6] [4 2 5 1 6 3]

结合代码。

  1. public List<Integer> inorderTraversal(TreeNode root) {
  2. List<Integer> ans = new ArrayList<>();
  3. Stack<TreeNode> stack = new Stack<>();
  4. TreeNode cur = root;
  5. while (cur != null || !stack.isEmpty()) {
  6. //节点不为空一直压栈
  7. while (cur != null) {
  8. stack.push(cur);
  9. cur = cur.left; //考虑左子树
  10. }
  11. //节点为空,就出栈
  12. cur = stack.pop();
  13. //当前值加入
  14. ans.add(cur.val);
  15. //考虑右子树
  16. cur = cur.right;
  17. }
  18. return ans;
  19. }

时间复杂度:O(n)。

空间复杂度:O(h),栈消耗,h 是二叉树的高度。

解法三 Morris Traversal

解法一和解法二本质上是一致的,都需要 O(h)的空间来保存上一层的信息。而我们注意到中序遍历,就是遍历完左子树,然后遍历根节点。如果我们把当前根节点存起来,然后遍历左子树,左子树遍历完以后回到当前根节点就可以了,怎么做到呢?

我们知道,左子树最后遍历的节点一定是一个叶子节点,它的左右孩子都是 null,我们把它右孩子指向当前根节点存起来,这样的话我们就不需要额外空间了。这样做,遍历完当前左子树,就可以回到根节点了。

当然如果当前根节点左子树为空,那么我们只需要保存根节点的值,然后考虑右子树即可。

所以总体思想就是:记当前遍历的节点为 cur。

1、cur.left 为 null,保存 cur 的值,更新 cur = cur.right

2、cur.left 不为 null,找到 cur.left 这颗子树最右边的节点记做 last

2.1 last.right 为 null,那么将 last.right = cur,更新 cur = cur.left

2.2 last.right 不为 null,说明之前已经访问过,第二次来到这里,表明当前子树遍历完成,保存 cur 的值,更新 cur = cur.right

结合图示:

94. Binary Tree Inorder Traversal - 图2

如上图,cur 指向根节点。 当前属于 2.1 的情况,cur.left 不为 null,cur 的左子树最右边的节点的右孩子为 null,那么我们把最右边的节点的右孩子指向 cur。

94. Binary Tree Inorder Traversal - 图3

接着,更新 cur = cur.left。

94. Binary Tree Inorder Traversal - 图4

如上图,当前属于 2.1 的情况,cur.left 不为 null,cur 的左子树最右边的节点的右孩子为 null,那么我们把最右边的节点的右孩子指向 cur。

94. Binary Tree Inorder Traversal - 图5

更新 cur = cur.left。

94. Binary Tree Inorder Traversal - 图6

如上图,当前属于情况 1,cur.left 为 null,保存 cur 的值,更新 cur = cur.right。

94. Binary Tree Inorder Traversal - 图7

如上图,当前属于 2.2 的情况,cur.left 不为 null,cur 的左子树最右边的节点的右孩子已经指向 cur,保存 cur 的值,更新 cur = cur.right。

94. Binary Tree Inorder Traversal - 图8

如上图,当前属于情况 1,cur.left 为 null,保存 cur 的值,更新 cur = cur.right。

94. Binary Tree Inorder Traversal - 图9

如上图,当前属于 2.2 的情况,cur.left 不为 null,cur 的左子树最右边的节点的右孩子已经指向 cur,保存 cur 的值,更新 cur = cur.right。

94. Binary Tree Inorder Traversal - 图10

当前属于情况 1,cur.left 为 null,保存 cur 的值,更新 cur = cur.right。

94. Binary Tree Inorder Traversal - 图11

cur 指向 null,结束遍历。

根据这个关系,写代码

记当前遍历的节点为 cur。

1、cur.left 为 null,保存 cur 的值,更新 cur = cur.right

2、cur.left 不为 null,找到 cur.left 这颗子树最右边的节点记做 last

2.1 last.right 为 null,那么将 last.right = cur,更新 cur = cur.left

2.2 last.right 不为 null,说明之前已经访问过,第二次来到这里,表明当前子树遍历完成,保存 cur 的值,更新 cur = cur.right

  1. public List<Integer> inorderTraversal(TreeNode root) {
  2. List<Integer> ans = new ArrayList<>();
  3. TreeNode cur = root;
  4. while (cur != null) {
  5. //情况 1
  6. if (cur.left == null) {
  7. ans.add(cur.val);
  8. cur = cur.right;
  9. } else {
  10. //找左子树最右边的节点
  11. TreeNode pre = cur.left;
  12. while (pre.right != null && pre.right != cur) {
  13. pre = pre.right;
  14. }
  15. //情况 2.1
  16. if (pre.right == null) {
  17. pre.right = cur;
  18. cur = cur.left;
  19. }
  20. //情况 2.2
  21. if (pre.right == cur) {
  22. pre.right = null; //这里可以恢复为 null
  23. ans.add(cur.val);
  24. cur = cur.right;
  25. }
  26. }
  27. }
  28. return ans;
  29. }

时间复杂度:O(n)。每个节点遍历常数次。

空间复杂度:O(1)。

解法三是自己第一次见到,充分利用原来的空间的遍历,太强了。这么好的算法,当时上课的时候为什么没有讲,可惜了。

windliang wechat

添加好友一起进步~

如果觉得有帮助的话,可以点击 这里 给一个 star 哦 ^^

如果想系统的学习数据结构和算法,强烈推荐一个我之前学过的课程,可以点击 这里 查看详情