257:二叉树的所有路径

问题描述

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

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

示例

输入:

   1
 /   \
2     3
 \
  5

输出: ["1->2->5", "1->3"]

解释: 所有根节点到叶子节点的路径为: 1->2->5, 1->3

思路

先序遍历。用一个字符串记录走过的路,遇到非叶子就加入串。遇到叶子结点,加入串后,要把整个串加入结果序列。(方法一)
方法一用了18ms才通过所有的case。考虑到字符串拼接的效率问题,用stringbuilder改造了一发,提速至2ms。(方法二)

方法一

Java版

class Solution {
    List<String> list = new ArrayList<>();
    public List<String> binaryTreePaths(TreeNode root) {
        if(root == null){
            return list;
        }
        binaryTree(root,"");
        return list;
    }
    private void binaryTree(TreeNode root, String s){
        s+=root.val;
        if(root.left == null && root.right == null){
            list.add(s);
            return;
        }
        s+="->";
        if(root.left != null){
            binaryTree(root.left,s);
        }
        if(root.right != null){
            binaryTree(root.right,s);
        }
    }

}

方法二

Java版

class Solution {
    List<String> list = new ArrayList<>();
    public List<String> binaryTreePaths(TreeNode root) {
        if(root == null){
            return list;
        }
        binaryTree(root,new StringBuilder());
        return list;
    }
    private void binaryTree(TreeNode root, StringBuilder s){
        s.append(root.val);
        if(root.left == null && root.right == null){
            list.add(s.toString());
            return;
        }
        s.append("->");
        if(root.left != null){
            binaryTree(root.left,new StringBuilder(s));
        }
        if(root.right != null){
            binaryTree(root.right,new StringBuilder(s));
        }
    }

}

发布了396 篇原创文章 · 获赞 22 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/weixin_41687289/article/details/104835876
今日推荐