力扣94 二叉树的中序遍历 (Java版本) 递归、非递归

文章目录

  • 题目描述
  • 递归解法
  • 非递归解法


题目描述

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

示例 1:
力扣94 二叉树的中序遍历 (Java版本) 递归、非递归_第1张图片

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

输入:root = []
输出:[]
示例 3:

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

提示:

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

进阶: 递归算法很简单,你可以通过迭代算法完成吗?

递归解法

class Solution {
    //使用递归的方法
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        inorder(root,result);
        return  result;
    }

    public void inorder(TreeNode root,List<Integer> result){
        if (root==null){
            return;
        }
        if (root.left!=null){
            inorder(root.left,result);
        }
        result.add(root.val);
        if(root.right!=null){
            inorder(root.right,result);
        }
    }

}

非递归解法

class Solution {
    //使用非递归的方法
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        inorder(root,result);
        return  result;
    }
    
    public void inorder(TreeNode root,List<Integer> result){
        if (root==null){
            return;
        }
        Stack<TreeNode> stack = new Stack<>();
        TreeNode cur=root;//负责遍历
        //cur一直往左孩子的方向走,并且把指向的节点压入栈中
        //如果cur当前指向的节点为空,就开始从栈中弹出一个节点并且遍历(弹出的这个节点相当于cur上一次指向的节点),遍历完这个节点之后就开始往右孩子的方向走
        while (cur!=null||!stack.isEmpty()){
            if (cur!=null){
                stack.push(cur);
                cur = cur.left;
            }else {
                //cur为空则表示上个节点没有左孩子,弹出刚才的节点并且遍历
                cur = stack.pop();
                result.add(cur.val);
                //开始往当前节点的右孩子方向走
                cur = cur.right;
                //然后结束本轮循环,进入下一轮循环。如果cur为空则表示也没有右孩子,就继续从栈顶弹出元素遍历,如果不为空则压入栈并且开始找左孩子。
            }
        }
    }
}

你可能感兴趣的:(LeetCode,leetcode,java)