题目:
给你一个二叉树的根节点 root
,按 任意顺序 ,返回所有从根节点到叶子节点的路径。
叶子节点 是指没有子节点的节点。
思路:
递归与回溯。使用前序遍历遍历整棵树。
第一步:确定参数与返回值,参数为节点,存放路径的List,存放结果的结果集result,递归函数无返回值。
第二步:确定结束条件。如果节点是叶子节点,存放路径的List中就存放了一整条路径,将其加入结果集。
第三步:确定单层递归逻辑。将节点加入List中,如果节点有左孩子,就对左孩子进行递归,再回溯;如果节点有右孩子,就对右孩子进行递归,再回溯。回溯就是删节点。
代码:
public List<String> binaryTreePaths(TreeNode root) {
List<String> result=new ArrayList<>();//最终的结果
if(root==null)
return result;
List<Integer> paths=new ArrayList<>();//结果中的路径
traversal(root,paths,result);
return result;
}
public void traversal(TreeNode root,List<Integer> paths,List<String> result){
paths.add(root.val);//处理中节点
//遇到叶子节点
if(root.left==null&&root.right==null){
StringBuilder s=new StringBuilder();
for(int i=0;i<paths.size()-1;i++){
s.append(paths.get(i)).append("->");
}
s.append(paths.get(paths.size()-1));//记录本条路径的最后一个节点
result.add(s.toString());//完成一条路径的遍历
}
//有左子树
if(root.left!=null){
traversal(root.left,paths,result);
paths.remove(paths.size()-1);//回溯
}
//有右子树
if(root.right!=null){
traversal(root.right,paths,result);
paths.remove(paths.size()-1);//回溯
}
}