题目
输入一颗二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始一直到树的叶子节点形成的路径。
解题思路
先序遍历: 本题我采用的是先序遍历树的所有节点,也就是按照根、左、右的顺序。
详细思路: 由于本题目是一个将二叉树对应的路径给输出,所有需要两个方法,一个是专门负责进行递推,将所有符合的节点都输出,另一个就是负责将这个结果列表进行一个返回。
两个方法:
pathSum(root,sum)函数:
- 初始化:结果列表res,路径列表path
- 返回值:返回res即可
recur(root,tar)函数
- 递推参数:当前节点root,当前目标值var
- 终止条件:若节点为空,则直接返回。
- 递推工作:
1.路径更新: 将当前节点值 root.val 加入路径 path ;
2.目标值更新: tar = tar - root.val(即目标值 tar 从 sum 减至 0);
3.路径记录: 当 ① root 为叶节点 且 ② 路径和等于目标值 ,则将此路径 path 加入 res 。
先序遍历: 递归左 / 右子节点。
4.路径恢复: 向上回溯前,需要将当前节点从路径 path 中删除,即执行 path.pop()
代码
class Solution {
LinkedList<List<Integer>> res = new LinkedList<>();//首先就是创建这样的一个包含右链表的链表,因为返回的可能有多个列表
LinkedList<Integer> path = new LinkedList<>();//这个就是单独派出的一个链表进行先序遍历
public List<List<Integer>> pathSum(TreeNode root, int sum) {
recur(root, sum);
return res;
}
void recur(TreeNode root, int tar) {
if(root == null) return;
path.add(root.val);
tar -= root.val;//这个就是用的减号,使得对应的东西不断减去,直到为0
if(tar == 0 && root.left == null && root.right == null)
res.add(new LinkedList(path));
recur(root.left, tar);
recur(root.right, tar);
path.removeLast();
}
}
总结
本题的重要思路就是用到了递归,通过第一次执行完成,判断其是否达到了减去之后为0,如果不为0,那么选择其他路径。