【遍历】非递归法二叉树的前中后序遍历

文章目录

非递归法

通过栈Stack来模拟递归。

前序遍历

LeetCode 144
在这里插入图片描述

前序遍历：1 2 3

定义：存放答案的List、栈Stack

将root入栈
出栈：node，为null则舍弃
将node放入list
将node.right入栈
将node.left入栈
栈不为空则重复2-5步

为了让左节点优先于右节点出栈，因此先将右节点入栈。

class Solution {
    
    
    public List<Integer> preorderTraversal(TreeNode root) {
    
    
        Stack<TreeNode> stack = new Stack<>();

        List<Integer> list = new LinkedList<>();
        stack.push(root);
        while(!stack.empty()){
    
    
            TreeNode node = stack.pop();
            if(node==null)continue;
            list.add(node.val);
            stack.push(node.right);
            stack.push(node.left);
        }
        return list;
    }
}

后序遍历

LeetCode 145
在这里插入图片描述

后序遍历：2 3 1

后序遍历仅需在前序遍历的代码中修改3处即可。

由前序遍历1 2 3 改为 1 3 2 再翻转为 2 3 1即为答案。

class Solution {
    
    
    public List<Integer> postorderTraversal(TreeNode root) {
    
    
        Stack<TreeNode> stack = new Stack<>();
        List<Integer> list = new LinkedList<>();
        stack.push(root);
        while(!stack.empty()){
    
    
            TreeNode node = stack.pop();
            if(node == null)continue;

            list.add(node.val);
            stack.push(node.left); // 先放入左节点
            stack.push(node.right); 
        }

        Collections.reverse(list); // 反转
        return list;
    }
}

中序遍历

LeetCode 94

中序遍历代码与前序和后续不同。

在这里插入图片描述

中序遍历： 4 2 5 1 3。

思考：要想先输出4，则需要将左节点持续入栈，直到为null，此时出栈即为4，然后将其右节点入栈…

同样的，定义存放结果的list和栈stack。

cur = root
cur不为空或者栈不为空
循环将cur入栈，并将cur赋值其左节点，直到为空
出站node，将node加入list
将node赋值为node.left
重复2 - 5步

class Solution {
    
    

    public List<Integer> inorderTraversal(TreeNode root) {
    
    
        
        Stack<TreeNode> stack = new Stack<>();
        List<Integer> list = new LinkedList<>();
        TreeNode cur = root;
        
        while(cur!=null||!stack.empty()){
    
    
            while(cur!=null){
    
    
                stack.push(cur);
                cur = cur.left;
            }
            TreeNode node = stack.pop(); 
            list.add(node.val);
            cur = node.right;
        }
        return list;
    }
}

递归法DFS

class Solution {
    
    
	List<Integer> list1 = new LinkedList<>(); // 前序
	List<Integer> list2 = new LinkedList<>(); // 中序
	List<Integer> list3 = new LinkedList<>(); // 后序

    public List<Integer> inorderTraversal(TreeNode root) {
    
    
			traverse(root);
			return list2; 
    }
	
	void traverse(TreeNode root){
    
    
		if(root==null)return;
		
		list1.add(root.val); 
		
		traverse(root.left); // 递归左节点
		
		list2.add(root.val);
		
		traverse(root.right); // 递归右节点
		
		list3.add(root.val);
		
	}
	
}

参考：

【遍历】非递归法 二叉树的前中后序遍历