LeetCode_257_二叉树的所有路径

题目链接

• https://leetcode.cn/problems/binary-tree-paths/

题目描述

给你一个二叉树的根节点 root ，按 任意顺序 ，返回所有从根节点到叶子节点的路径。

叶子节点 是指没有子节点的节点。

示例 1：

输入：root = [1,2,3,null,5]
输出：["1->2->5","1->3"]

示例 2：

输入：root = [1]
输出：["1"]

提示：

• 树中节点的数目在范围 [1, 100] 内
• -100 <= Node.val <= 100

解题思路

求根节点到叶子的路径，所以需要前序遍历，这样才方便让父节点指向孩子节点，找到对应的路径

递归法

• 明确递归函数的参数和返回值
  • 参数：要传入的根节点root；记录每一条路径的path；存放结果的ans
  • 返回值：空
• 明确终止条件
  • 找到叶子节点
• 明确单层逻辑
  • 因为是前序遍历，需要先处理中间节点，中间节点就是要记录路径上的节点，先放进path中
  • 然后是递归和回溯的过程，递归前要判断一下下面要递归的节点是否为空
  • 递归完还要做回溯，因为path不能一直加入节点，他还要删除节点，然后才能加入新的节点

AC代码

递归法

class Solution {
    public List<String> binaryTreePaths(TreeNode root) {
        List<String> ans = new ArrayList<>();
        if (root == null) {
            return ans;
        }
        List<Integer> path = new ArrayList<>();
        traversal(root, path, ans);
        return ans;
    }

    private static void traversal(TreeNode root, List<Integer> path, List<String> ans) {
        path.add(root.val);
        //叶子节点
        if (root.left == null && root.right == null) {
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < path.size() - 1; i++) {
                sb.append(path.get(i)).append("->");
            }
            sb.append(path.get(path.size() - 1));
            ans.add(sb.toString());
            return;
        }
        if (root.left != null) {
            traversal(root.left, path, ans);
            path.remove(path.size() - 1);
        }
        if (root.right != null) {
            traversal(root.right, path, ans);
            path.remove(path.size() - 1);
        }
    }
}