问题描述
给定一个二叉树,返回所有从根节点到叶子节点的路径。
说明: 叶子节点是指没有子节点的节点。
示例
输入:
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));
}
}
}