找回密码
 立即注册
首页 业界区 科技 二叉树的中序遍历:递归法和迭代法

二叉树的中序遍历:递归法和迭代法

窖咎 昨天 17:52
题目:给定一个二叉树的根节点 root ,返回 它的 中序 遍历 。
1.png

对于中序遍历,两种最常用的方法是递归法和迭代法。下面分别给出这两种方法最简洁的 Java 实现,任选其一即可。
中序遍历顺序:左子树 → 根节点 → 右子树
递归法最直观,而迭代法用栈避免递归深度问题。
复杂度:

  • 时间:O(n)
  • 空间:递归 O(h),迭代 O(h)(h = 树高)
一、递归法的 Java 实现:
  1. /**
  2. * Definition for a binary tree node.
  3. * public class TreeNode {
  4. *     int val;
  5. *     TreeNode left;
  6. *     TreeNode right;
  7. *     TreeNode() {}
  8. *     TreeNode(int val) { this.val = val; }
  9. *     TreeNode(int val, TreeNode left, TreeNode right) {
  10. *         this.val = val;
  11. *         this.left = left;
  12. *         this.right = right;
  13. *     }
  14. * }
  15. */
  16. class Solution {
  17.     public List<Integer> inorderTraversal(TreeNode root) {
  18.         if(root==null){
  19.             // 注意返回结果要List类型,不能 return null
  20.             return new ArrayList();
  21.         }
  22.         List<Integer> res = new ArrayList<>();
  23.         inorder(root,res);
  24.         return res;
  25.     }
  26.     public void inorder(TreeNode root, List<Integer> res){
  27.         // 注意 左结点或右结点为空,就回到上一层
  28.         if(root==null){
  29.             return;
  30.         }
  31.         inorder(root.left, res);   //左
  32.         res.add(root.val);         //根
  33.         inorder(root.right, res);  //右
  34.     }
  35. }
复制代码
二、迭代法的 Java 代码:
  1. class Solution {
  2.     public List<Integer> inorderTraversal(TreeNode root) {
  3.         List<Integer> res = new ArrayList<>();
  4.         Deque<TreeNode> stack = new ArrayDeque<>();
  5.         TreeNode cur = root;
  6.         while (cur != null || !stack.isEmpty()) {
  7.             while (cur != null) {      // 一路向左
  8.                 stack.push(cur);
  9.                 cur = cur.left;
  10.             }
  11.             cur = stack.pop();         // 当前子树最左节点
  12.             res.add(cur.val);          // 访问根
  13.             cur = cur.right;           // 转向右子树
  14.         }
  15.         return res;
  16.     }
  17. }
复制代码
来源:程序园用户自行投稿发布,如果侵权,请联系站长删除
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!
您需要登录后才可以回帖 登录 | 立即注册