【leetcode】第六章 二叉树part01

递归遍历

144. 二叉树的前序遍历

java 复制代码
// 前序遍历
public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        preOrder(root, res);
        return res;
    }

    private void preOrder(TreeNode root, List<Integer> res) {
        if (root == null) return;
        
        res.add(root.val);
        preOrder(root.left, res);
        preOrder(root.right, res);
    }

94. 二叉树的中序遍历

java 复制代码
public List<Integer> inorderTraversal(TreeNode root) {
        // 递归
        List<Integer> res = new ArrayList<>();
        inOrder(root, res);
        return res;
    }

    private void inOrder(TreeNode root, List<Integer> res) {
        if (root == null) return;

        inOrder(root.left, res);
        res.add(root.val);
        inOrder(root.right, res);
    }

145. 二叉树的后序遍历

java 复制代码
public List<Integer> postorderTraversal(TreeNode root) {
        // 递归
        List<Integer> res = new ArrayList<>();
        postOrder(root, res);
        return res;

    }

    private void postOrder(TreeNode root, List<Integer> res) {
        if (root == null) return;

        postOrder(root.left, res);
        postOrder(root.right, res);
        res.add(root.val);

    }

非递归遍历

  • 前序
Java 复制代码
public List<Integer> preorderTraversal(TreeNode root) {
        // 非递归
        List<Integer> res = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        while (root != null || !stack.isEmpty()) {
            // 一直将左节点压入栈中
            while (root != null) {
                res.add(root.val); // 根节点首先访问
                stack.push(root);
                root = root.left;
            }
            root = stack.pop();
            root = root.right; 
        }

        return res;
    }
  • 中序
Java 复制代码
public 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;
            }

            root = stack.pop();
            res.add(root.val);
            root = root.right;
        }
        return res;
    }
  • 后序
Java 复制代码
public List<Integer> postorderTraversal(TreeNode root) {
    // 非递归
    List<Integer> res = new ArrayList<>();
    Stack<TreeNode> stack = new Stack<>();
    TreeNode prev = null;
    while (root != null || !stack.isEmpty()) {
        // 将左节点压入栈
        while (root != null) {
            stack.push(root);
            root = root.left;
        }

        root = stack.pop();
        // 后序遍历的过程中在遍历完左子树跟右子树cur都会回到根结点。
        // 所以当前不管是从左子树还是右子树回到根结点都不应该再 操作了,应该退回上层。
        // 如果是从右边再返回根结点,应该回到上层。
        // 主要就是判断出来的是不是右子树,是的话就可以把根节点=加入到list了
        if (root.right == null || root.right == prev) {
            res.add(root.val);
            prev = root;
            root = null;
        }
        else {
            stack.push(root);
            root = root.right;
        }

    }

    return res;

}
相关推荐
Joyner20186 小时前
python-leetcode-从中序与后序遍历序列构造二叉树
算法·leetcode·职场和发展
因兹菜6 小时前
[LeetCode]day9 203.移除链表元素
算法·leetcode·链表
LNsupermali6 小时前
力扣257. 二叉树的所有路径(遍历思想解决)
算法·leetcode·职场和发展
雾月556 小时前
LeetCode LCR180文件组合
算法·leetcode·职场和发展
萌の鱼6 小时前
leetcode 2080. 区间内查询数字的频率
数据结构·c++·算法·leetcode
Tisfy6 小时前
LeetCode 0541.反转字符串 II:模拟
算法·leetcode·字符串·题解
labmem19 小时前
Leetcode:541
算法·leetcode·职场和发展
圆圆滚滚小企鹅。9 小时前
刷题记录 HOT100回溯算法-6:79. 单词搜索
笔记·python·算法·leetcode
钓一朵雪12 小时前
【Leetcode刷题记录】45. 跳跃游戏 II--贪心算法
leetcode·贪心算法
sjsjs1112 小时前
【数据结构-前缀树】力扣208. 实现 Trie (前缀树)
数据结构·leetcode