0
点赞
收藏
分享

微信扫一扫

二叉树前中后序遍历(循环实现)

陆公子521 2022-02-07 阅读 49
深度优先

前序遍历

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

中序遍历

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

        return res;
    }
}

后序遍历

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        
        while(root != null || !stack.isEmpty()){
            if(root != null){
                stack.push(root);
                res.add(0,root.val);
                root = root.right;
            }else{
                root = stack.pop();
                root = root.left;
            }
        }

        return res;
    }
}
举报

相关推荐

0 条评论