LeetCode94二叉树的中序遍历(递归与迭代)

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

LeetCode94二叉树的中序遍历(递归与迭代)_第1张图片

示例 1:

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

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

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

中序遍历的特点就是 <先遍历左子树 再遍历根 再遍历右子树>

如图
LeetCode94二叉树的中序遍历(递归与迭代)_第2张图片

思路1 递归思想

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);
    }

迭代

LeetCode94二叉树的中序遍历(递归与迭代)_第3张图片

public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        TreeNode head = root;
        while(head!=null){
            stack.push(head);
            head = head.left;
        }
        while(!stack.isEmpty()){
            //出栈顶     
            //到了子树的根 判断根是否有右子树,如果有,那就一直入栈
            TreeNode temp = stack.pop();
            list.add(temp.val);
            if(temp.right!=null){
                temp = temp.right;
                while(temp!=null){
                    stack.push(temp);
                    temp = temp.left;
                }
            }
            //入栈完继续重复上述
        }
        return list;
    }

你可能感兴趣的:(数据结构,java,算法题,算法,java,数据结构)