问题描述:找出一个二叉树中,根节点到所有叶子节点的路径。
思路:如果节点左右子树都不存在,表明为叶子节点。将路径保存并返回。只是根节点就是叶子节点时,要特殊判断下。
原答案:
private List<String> resList=new ArrayList<String>();
public List<String> binaryTreePaths(TreeNode root) {
List<String> list=new ArrayList<String>();
if(root==null)
return list;
String path=String.valueOf(root.val);
if(root.left==null&&root.right==null){
resList.add(path);
return resList;
}
if(root.left!=null)
binaryTreePaths(root.left,path);
if(root.right!=null)
binaryTreePaths(root.right,path);
return resList;
}
private void binaryTreePaths(TreeNode root,String path){
path=path+"->"+root.val;
if(root.left==null&&root.right==null){
resList.add(path);
return;
}
if(root.left!=null)
binaryTreePaths(root.left,path);
if(root.right!=null)
binaryTreePaths(root.right,path);
return;
}
最佳答案:
public List<String> binaryTreePaths(TreeNode root) {
List<String> paths = new ArrayList<>();
binaryTreePaths(root, "", paths);
return paths;
}
private void binaryTreePaths(TreeNode root, String prefix, List<String> paths) {
if (root == null) return;
if (root.left == null && root.right == null) {
paths.add(prefix + root.val);
return;
}
String path = prefix + root.val + "->";
binaryTreePaths(root.left, path, paths);
binaryTreePaths(root.right, path, paths);
}
思路是一致的。但是对开始路径的处理就避免了根节点就是叶子节点的特殊处理。