LeetCode 94二叉树的中序遍历

题目:

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

递归方法实现
迭代方法实现

示例1:

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

示例2:

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

思路:
这是一道简单题,就是二叉树的中序遍历,中序遍历就是左子树——>根节点——>右子树的遍历顺序。不过注意示例二,如果二叉树为空返回的不是null。

违背开发习惯的递归代码:

class Solution {
    
    
    //定义一个链表接受数据
    List<Integer> list = new ArrayList<>();

    public List<Integer> inorderTraversal(TreeNode root) {
    
    
		//边界条件判断
        if (root == null) {
    
    
            return new ArrayList();
        }
        //先遍历左子树
        inorderTraversal(root.left);
        //将根节点值放入链表
        list.add(root.val);
        //再遍历右子树
        inorderTraversal(root.right);
        return list;
    }
}

符合开发习惯的代码:

class Solution {
    
    
    public List<Integer> inorderTraversal(TreeNode root) {
    
    
        List<Integer> res = new ArrayList<Integer>();
        inorder(root, res);
        return res;
    }

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

迭代方式实现中序遍历:
中序遍历迭代方式实现逻辑如下,很巧妙的逻辑。自己画个二叉树跟着代码debug一遍就可以理解。

class Solution {
    
    
    public List<Integer> inorderTraversal(TreeNode root) {
    
    
        List<Integer> res = new ArrayList<Integer>();
        Deque<TreeNode> stk = new LinkedList<TreeNode>();
        while (root != null || !stk.isEmpty()) {
    
    
            while (root != null) {
    
    
                stk.push(root);
                root = root.left;
            }
            root = stk.pop();
            res.add(root.val);
            root = root.right;
        }
        return res;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_43744992/article/details/121589455