【Leetcode刷题】二叉树的前序、中序、后续遍历（非递归）【Java实现】

最新推荐文章于 2024-11-10 22:49:00 发布

HaoTianYan

最新推荐文章于 2024-11-10 22:49:00 发布

阅读量205

点赞数

分类专栏：算法与数据结构 LeetCode 文章标签：数据结构

本文链接：https://blog.csdn.net/weixin_41804998/article/details/115677562

版权

算法与数据结构同时被 2 个专栏收录

8 篇文章 0 订阅

订阅专栏

LeetCode

5 篇文章 0 订阅

订阅专栏

本文详细介绍了如何使用非递归方法实现二叉树的前序、中序和后序遍历。通过栈来辅助遍历，避免递归带来的额外开销。在前序遍历中，先存节点值，然后遍历左子树；中序遍历不存节点值，直接遍历左子树；后序遍历则先插入节点值，遍历右子树后再回溯到左子树。

摘要由CSDN通过智能技术生成

0. 整体思路

用非递归的方法解决二叉树的三种遍历问题，主要思想其实就是利用了 【栈】 ，将暂时用不到的TreeNode节点存起来，先沿着某一分支（left或者right）走下去，等到了叶子节点（root==null）时再依次pop出来赋给root。

二叉树结点TreeNode的定义：

/**
 * Definition for a binary tree node.
 */
public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode() {}
    TreeNode(int val) { this.val = val; }
    TreeNode(int val, TreeNode left, TreeNode right) {
        this.val = val;
        this.left = left;
        this.right = right;
    }
}

1. 二叉树的前序遍历（非递归）

Leetcode：144. 二叉树的前序遍历

/**
 * Binary tree preorder traversal.
 */
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Deque<TreeNode> stack = new ArrayDeque<>();
        while (root != null || !stack.isEmpty()) {
        	// 先存入当前节点的val，再沿着左子树遍历
            if (root != null) {
                list.add(root.val);
                stack.push(root);
                root = root.left;
            } else {
            	// 遍历到头了则回溯，取上一个节点的右子树
                TreeNode temp = stack.pop();
                root = temp.right;
            }
        }
        return list;
    }
}

2. 二叉树的中序遍历（非递归）

Leetcode：94. 二叉树的中序遍历

/**
 * Binary tree inordertraversal.
 */
class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Deque<TreeNode> stack = new ArrayDeque<>();
        while (root != null || !stack.isEmpty()) {
        	// 不存入val，先沿着左子树遍历
            if (root != null) {
                stack.push(root);
                root = root.left;
            } else {
            	// 到叶子节点了，存入val，再利用stack回溯到上一个节点，取其右子树作为root
                TreeNode temp = stack.pop();
                list.add(temp.val);
                root = temp.right;
            }
        }
        return list;
    }
}

3. 二叉树的后序遍历（非递归）

Leetcode：145. 二叉树的后序遍历

后序遍历其实跟前序遍历非常像，不同点在于：

每次存入 list 时，都不是插入到末尾，而是用list.add(0,root.val)插入到 list 的最前面。
按照我们每次都插到最开头的做法，那么就要先遍历右子树，再遍历左子树了。

/**
 * Binary tree postorder traversal.
 */
 class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        Deque<TreeNode> stack = new ArrayDeque<>();
        while (root != null || !stack.isEmpty()) {
        	// 插入到list的最前方，然后先遍历右子树
            if (root != null) {
                list.add(0, root.val);
                stack.push(root);
                root = root.right;
            } else {
            	// 再用栈进行回溯，遍历左子树
                TreeNode temp = stack.pop();
                root = temp.left;
            }
        }
        return list;
    }
}