Leetcode 144.145.94.三种二叉树遍历通用方法(迭代) 记录反思

一般的二叉树迭代方法先序后序一样,中序不同,现在提供统一写法

因为迭代法使用栈去实现的,所以,先要打印的后入栈

和普通迭代法的区别就是在push当前节点的时候同时会添加一个null值作为标记

//中序
class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Deque<TreeNode> que = new LinkedList<>();
        
        if(root != null)
            que.push(root);

        while(!que.isEmpty()){
            TreeNode node = que.peek();
            if(node != null){
                que.pop();
                if(node.right != null)
                    que.push(node.right);//右子节点入栈,因为是最后打印
                que.push(node);//中间节点
                que.push(null);
                if(node.left != null)//左子节点入栈
                    que.push(node.left);
            }else{
                que.pop();
                node = que.peek();
                que.pop();
                list.add(node.val);
            }
        }
        return list;
    }
}

//先序
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
       List<Integer> list = new ArrayList<>();
        Deque<TreeNode> que = new LinkedList<>();
        
        if(root != null)
            que.push(root);

        while(!que.isEmpty()){
            TreeNode node = que.peek();
            if(node != null){
                que.pop();
                if(node.right != null)//右子节点
                    que.push(node.right);
                if(node.left != null)//左子节点
                    que.push(node.left);
                que.push(node);//中间节点
                que.push(null);
            }else{
                que.pop();
                node = que.peek();
                que.pop();
                list.add(node.val);
            }
        }
        return list;
    }
}
//后序
class Solution {
    
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Deque<TreeNode> que = new LinkedList<>();
        
        if(root != null)
            que.push(root);

        while(!que.isEmpty()){
            TreeNode node = que.peek();
            if(node != null){
                que.pop();
                que.push(node);//中间节点
                que.push(null);
                if(node.right != null)//右子节点
                    que.push(node.right);
                if(node.left != null)左子节点
                    que.push(node.left);
                
            }else{
                que.pop();
                node = que.peek();
                que.pop();
                list.add(node.val);
            }
        }
        return list;
    }
}