使用递归和回溯法找二叉树的所有路径
题目是 257.二叉树的所有路径
这道题需要我们找到所有从根节点到叶子结点的路径,起初我想的是使用深度优先遍历,但是我在遍历的时候没有找到方法来保存之前已经遍历过的节点。
按照代码随想录的解法,应该用一个List path保存之前已经遍历过的节点,当遍历到一个叶子节点就停止遍历,然后本轮递归的上一层递归把path的最后一个元素删掉。
如果当前节点不是叶子节点,就在递归当前节点的子节点后使用回溯,回溯的操作是删除当前节点的直接子节点,也就是path的最后一个元素,为什么只需要删除最后一个元素,因为在递归过程中已经删除了当前节点的其他子节点。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
private void myPath(List<String> list, List<Integer> path, TreeNode root){
path.add(root.val);
if(root.left==null && root.right==null){
StringBuilder sb=new StringBuilder();
for(int i=0; i<path.size()-1; i++){
sb.append(path.get(i));
sb.append("->");
}
sb.append(path.get(path.size()-1));
list.add(sb.toString());
return;
}
if(root.left!=null){
myPath(list, path, root.left);
//回溯到当前的节点
path.remove(path.size()-1);
}
if(root.right!=null){
myPath(list, path, root.right);
//回溯到当前的节点
path.remove(path.size()-1);
}
}
public List<String> binaryTreePaths(TreeNode root) {
StringBuilder sb;
List<String> list=new ArrayList<>();
List<Integer>path=new ArrayList<>();
myPath(list,path,root);
return list;
}
}