二叉树全部路径
今天在力扣上看到一道题,找出二叉树中所有的路径再返回。
题目如下:
给定一个二叉树,返回所有从根节点到叶子节点的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
输入:
1
/ \
2 3
\
5
输出: ["1->2->5", "1->3"]
解释: 所有根节点到叶子节点的路径为: 1->2->5, 1->3
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-paths
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题意已经很清晰了,从根节点到叶子节点算一条路径,读题的时候没看到,草草看了一眼就开始干,结果自己分析了半天,最终大约猜测是这么回事... ...也是够了... ...
一看到二叉树就想到递归来解,这道题显然需要遍历二叉树所有节点,叶子节点的特征也很明显,那就是左右子节点都为null,这也是结束递归的条件。
看了一下在线ide给出的方法返回值为List<String>,这道题在运算过程中应该是插入和删除比较频繁,所以我选择了LinkedList。
由于需要拼接字符串,我开始的设想是传一个StringBuilder下去,但是发现实现起来比较麻烦,而且完全没有这种必要,只要在查到叶子节点时,进行构建字符串即可。
所以我选择了传一个列表作为递归函数的参数,每一次向下递归都将节点插入列表,每一次返回上一层递归都减去刚才插入的节点,这样不会乱,当遇到叶子节点时,遍历列表构建字符串,然后再清空StringBuilder。
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { List<String> results = new LinkedList<>(); StringBuilder builder = new StringBuilder(); public List<String> binaryTreePaths(TreeNode root) { if (root == null) return results; findPath(root, new LinkedList<>()); return results; } public void findPath(TreeNode root, List<Integer> res) { res.add(root.val); if (root.left != null) { findPath(root.left, res); } if (root.right != null) { findPath(root.right, res); } if (root.left == null && root.right == null) { for (int i : res) { if (builder.length() == 0) { builder.append(i); } else { builder.append("->") .append(i); } } results.add(builder.toString()); builder.delete(0, builder.length()); } res.remove(res.size() - 1); } }
接下来考虑完正常情况就要思考特殊情况了,如果传了一个空的二叉树该怎么处理?这一点我在递归前做了判断,直接返回空的list即可。