二叉树的遍历迭代算法实现

最新推荐文章于 2023-02-17 21:04:31 发布

原创最新推荐文章于 2023-02-17 21:04:31 发布 · 157 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树

算法专栏收录该内容

2 篇文章

订阅专栏

本文深入探讨了二叉树这一核心数据结构，详细解析了前序遍历和两种中序遍历算法的实现，提供了清晰的代码示例，帮助读者理解和掌握二叉树遍历的原理与实践。

二叉树的数据结构：

    public static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }
    }

前序遍历

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

中序遍历

    private static List<Integer> inOrderTraversal(TreeNode root) {
        Stack<TreeNode> stack = new Stack<>();
        List<Integer> res = new ArrayList<>();
        do {
            while (root != null) {
                stack.push(root);
                root = root.left;
            }
            if (!stack.isEmpty()) {
                res.add(stack.peek().val);
                root = stack.pop().right;
            }
        } while (!stack.isEmpty() || root != null);
        return res;
    }

中序遍历二（20191123）

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