LeetCode - 257. 二叉树的所有路径

题目

给定一个二叉树，返回所有从根节点到叶子节点的路径。

思路

DFS 搜索，参数除了子节点外，再传入当前节点的路径和结果 List

复杂度分析

假设树有 n 个节点。

• 时间复杂度O(n)，遍历整棵树。
• 空间复杂度O(n)，空间复杂度与整棵树的节点数相关。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public static List<String> binaryTreePaths(TreeNode root) {
        List<String> paths = new ArrayList<>();
        dfs(root, "", paths);
        return paths;
    }

    private static void dfs(TreeNode root, String path, List<String> paths) {
        if (root == null) {
            return;
        }
        StringBuilder pathSB = new StringBuilder(path);
        pathSB.append(root.val);
        if (root.left == null && root.right == null) {
            paths.add(pathSB.toString());
        } else {
            pathSB.append("->");
            dfs(root.left, pathSB.toString(), paths);
            dfs(root.right, pathSB.toString(), paths);
        }
    }
}