题目
解题思路
使用回溯的思想,核心在保存上一个值的路径,采用递归的方式实现DFS,完成对整个路径的遍历。
使用的数据结构:一个String类型的ArrayList用来接收最后的结果,一个Integer类型的ArraList用来接收每一轮遍历到的节点的数值
主要思路:
- 当节点为null直接返回定义的res(此时为null)
- 当根节点不为null,进入递归中,递归的主要思想:
- base case :如果节点的左右子树均为null,此时代表已经走到了叶子节点,此时可以将之前存储的走过的每个节点的值加上“->”后存入到
一个StringBuffer中去(这里不要使用String,因为String是不可以进行修改的),采用for循环将所有元素存储完毕后,将最后的sb转为String
类型存入result数组中,然后退出函数。 - 单层递归逻辑:如果该节点不是叶子节点且该节点的左子树存在,则继续向下递归,将每次递归的节点值存入数组,到达最后的叶子节点时,将所有的
路径存储同时返回上一个节点,从而达到回溯的效果。右子树的操作一致。
代码
/**
* 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 {
public List<String> binaryTreePaths(TreeNode root) {
List<String> res = new ArrayList<>();
if (root == null) {
return res;
}
List<Integer> paths = new ArrayList<>();
Traversal(root, paths, res);
return res;
}
// 定义遍历函数
public void Traversal(TreeNode root, List<Integer> paths, List<String> result) {
paths.add(root.val);
// 判断到达叶子节点的情况
if (root.left == null && root.right == null) {
// 输出
StringBuffer sb = new StringBuffer();
for (int i = 0; i < paths.size() - 1; i++) {
sb.append(paths.get(i)).append("->");
}
sb.append(paths.get(paths.size() - 1));
result.add(sb.toString());
return;
}
// 如果不是叶子节点,则继续往下进行遍历,DFS思想
if (root.left != null) {
// 如果左子树不为0,进行递归
Traversal(root.left, paths, result);
// 移除最后一个元素实现回溯过程
paths.remove(paths.size() - 1);
}
if (root.right != null) {
Traversal(root.right, paths, result);
paths.remove(paths.size() - 1);// 进行回溯
}
}
}