力扣:二叉树的所有路径java
思路
使用先序遍历。因为要先将节点输入到paths中,在进行操作。
回溯。在节点返回返回上一级后,要进行回退操作,也就将path的最后个节点删除,实现回退。
流程
1.创建List结果集,和paths集记录走过的节点,通过paths的add()和remove(),实现回溯效果。
2.递归函数:递归三部曲:
参数和返回值(此处不需要返回值,结果直接输出了,参数要传递根节点,和paths集和List结果集)
结束条件:遇见叶子节点,输出该路径(节点的左右孩子为空时为叶子节点)
单层递归逻辑:
对当前节点输入paths(此操作要在结束条件之前,否则叶子节点无法进入paths)
判断左右孩子是否为空,不为空则进行递归操作,并进行回退操作,也就是弹出paths的最后一个元素,此时最后一个元素为叶子节点,所有要弹出。
代码
class Solution {
public List<String> binaryTreePaths(TreeNode root) {//主函数
List<Integer> path = new ArrayList<>();//存储节点的路径
List<String> result = new ArrayList<>();//结果集
traversal(root,path,result);//调用递归函数
return result;
}
public void traversal(TreeNode node,List<Integer> path,List<String> result){//参数为根节点和总路径path和结果集
path.add(node.val);//先序操作的“中”,要放在结束条件之前,这样才能将叶子节点也输入到路径path中,否则叶子节点被忽略
if(node.left ==null && node.right == null){//结束条件,节点为根节点时输出路径
StringBuilder Spath = new StringBuilder();//存储路径String加上->符号
int i=0;
for(;i<path.size()-1;i++){//加到最后第二个
Spath.append(path.get(i));
Spath.append("->");
}
Spath.append(path.get(i));//最后一个后面不加->
result.add(Spath.toString());//将结果输入到结果集中
return;
}
if(node.left!=null){//要判定左节点不为空,否自会出现空指针错误,
traversal(node.left,path,result);//再递归调用该节点的左孩子
path.remove(path.size()-1);//回溯操作,将左叶子节点从路径表中删除(此时path包含根节点到该节点的左叶子节点,既然要回退就要删除它),实现回退
}
if(node.right!=null){
traversal(node.right,path,result);
path.remove(path.size()-1);
}
}
}