二叉树全部路径

今天在力扣上看到一道题,找出二叉树中所有的路径再返回。

题目如下:

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

说明: 叶子节点是指没有子节点的节点。

示例:

输入:

   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即可。

posted @ 2020-11-30 14:52  无心大魔王  阅读(263)  评论(0编辑  收藏  举报