「一分钟学算法」二叉树（一）

二叉树的中序遍历

*************

解法一

预备知识：递归，遍历顺序：左-根-右

/**
 * 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;
 *     }
 * }
 */
class Solution {
    LinkedList<Integer> res = new LinkedList<>();
    public List<Integer> inorderTraversal(TreeNode root) {
        if (root == null) {
            return new LinkedList<>();
        }
        inorderTraversal(root.left);
        res.add(root.val);
        inorderTraversal(root.right);
        return res;
    }
}

提交验证

解法二

预备知识：栈的特性：先进后出

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

提交验证

引申

二叉树前序遍历

顺序：根-左-右

代码稍作改动：

    public List<Integer> preorderTraversal(TreeNode root) {
        if (root != null) {
            res.add(root.val);//先跟节点
            preorderTraversal(root.left); //递归左节点
            preorderTraversal(root.right);//递归右节点
        }
        return res;
    }

二叉树后序遍历

顺序：左-右-根

代码稍作改动：

    public List<Integer> postorderTraversal(TreeNode root) {
        if (root != null) {
            postorderTraversal(root.left);
            postorderTraversal(root.right);
            res.add(root.val);
        }
        return res;
    }

总结

前序/中序/后序，遍历对应的根节点为前/中/后，子树均为先左后右。

递归方案解法容易理解，Stack栈方案比较绕。性能相似。或者说能用Stack栈解决的方案均可以用递归算法处理。

涉及算法：递归，数据结构：栈

作用

• 前序遍历
  • 输出某个文件夹下所有文件名及子文件夹
  • 获取树的最大深度
• 后序遍历
  • 统计某个文件夹的大小（该文件夹下所有文件的大小）