算法小抄学习笔记 — 8.二叉树的遍历

先序遍历

void preOrder(TreeNode root) {
    if (root == null) return;
    
    visited(root);			// 先序遍历位置
    preOrder(root.left);
    preOrder(root.right);
}

中序遍历

void inOrder(TreeNode root) {
    if (root == null) return;
    
    preOrder(root.left);
    visited(root);		// 中序遍历位置
    preOrder(root.right);
}

后序遍历

void postOrder(TreeNode root) {
    if (root == null) return;
    
    preOrder(root.left);
    preOrder(root.right);
    visited(root);		// 后序遍历位置
}

层次遍历（1）

单纯的访问遍历节点

void levelOrder(TreeNode root) {
    Queue<TreeNode> q = new LinkedList<>();
    if (root != null) {
        q.offer(root);
    }
    while (!q.isEmpty()) {
        TreeNode node = q.poll();
        visited(node);		// 层次遍历位置
        if (node.left != null) {
            q.offer(node.left);
        }
        if (node.right != null) {
            q.offer(node.right);
        }
    }
}

层次遍历（2）

需要知道访问节点在哪一层时。

例如：https://leetcode-cn.com/problems/binary-tree-level-order-traversal/submissions/

void levelOrder(TreeNode root) {
    Queue<TreeNode> q = new LinkedList<>();
    if (root != null) {
        q.offer(root);
    }
    while (!q.isEmpty()) {
        int size = q.size();
        for (int i = 0; i < size; i++) {	// 这儿不要用q.size()
            TreeNode node = q.poll();
            visited(node);		// 层次遍历位置
            if (node.left != null) {
                q.offer(node.left);
            }
            if (node.right != null) {
                q.offer(node.right);
            }
        }
    }
}