代码随想录-刷题笔记
内容:
本题是一道递归+回溯的问题,其实递归和回溯是一个意思,但是这道题一定要明显的进行一次回溯。
毕竟每一条路径都不太一样,收集完一次结果之后,需要把上一次探的节点给撤销.这一步就是很明显回溯
回溯部分在回溯篇章再去详细写一篇笔记,因此本文只做介绍,至于路径问题,真可谓是回溯的常客了.
哦,对每次都想吐槽一下Java的神奇的字符串,必须要用StringBuilder才可以.
/**
* 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 {
List<String> result = new ArrayList<>();
List<Integer> path = new ArrayList<>();
public void dfs(TreeNode root) {
if(root.left == null && root.right == null) {
path.add(root.val);
StringBuilder sb = new StringBuilder();
for(int i = 0 ; i < path.size()-1 ; i++) {
sb.append(String.valueOf(path.get(i)));
sb.append("->");
}
sb.append(String.valueOf(path.getLast()));
result.add(sb.toString());
path.removeLast(); //每次收集完一次路径,因为判断是叶子节点,所以需要手动来remove
return;
}
if(root.left != null) {
path.add(root.val);
dfs(root.left);
path.removeLast(); //回溯
}
if(root.right != null) {
path.add(root.val);
dfs(root.right);
path.removeLast(); //回溯
}
}
public List<String> binaryTreePaths(TreeNode root) {
if(root != null) dfs(root);
return result;
}
}
这里实际上每个节点都载入两次,可以仔细看一下代码逻辑
总结:
该题是回溯的一个启蒙题,只要把 后面两个if判断改成for的多枝,即为回溯模板