[Leetcode]144. Binary Tree Preorder Traversal

最新推荐文章于 2024-05-30 07:40:02 发布

原创最新推荐文章于 2024-05-30 07:40:02 发布 · 200 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

本文介绍了解决LeetCode上二叉树前序遍历问题的两种方法：递归和迭代。递归方法通过深度优先搜索的方式，先访问根节点，再分别访问左子树和右子树；迭代方法则利用栈来实现同样的过程，避免了递归可能带来的堆栈溢出问题。

link: https://leetcode.com/problems/binary-tree-preorder-traversal/

Solution1: 使用递归

class Solution {
    public void helper(TreeNode root, List<Integer> res) {
        if(root != null) res.add(root.val);
        if(root.left != null) helper(root.left, res);
        if(root.right != null) helper(root.right, res);
    }
    
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if(root == null) return res;
        helper(root, res);
        return res;
        
    }
}

TC: O()

Solution2: 使用迭代

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        Stack stack = new Stack();
        if(root == null) return res;
        
        while(root != null || !stack.empty()){
            if(root == null) {
                root = ((TreeNode)stack.pop()).right;
                continue;
            }
            res.add(root.val);
            stack.push(root);
            root = root.left;
        }
        return res;
    }
}