一、题目
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
二、实现
/**
* 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 {
LinkedList<List<Integer>> res = new LinkedList<>();
LinkedList<Integer> path = new LinkedList<>();
public List<List<Integer>> pathSum(TreeNode root, int target) {
getPath(root,target);
return res;
}
public void getPath(TreeNode root, int target){
if(root == null) return; // 当前树为空
path.add(root.val); // 将当前节点值加入路径
target = target - root.val; // 目标值减去加入的节点值
if(target == 0 && root.left == null && root.right == null){ // 当前路径和与目标值一致
res.add(new LinkedList(path));
}
getPath(root.left,target);
getPath(root.right,target);
path.removeLast(); //回溯
}
}
三、拓展题目
给你二叉树的根节点 root 和一个表示目标和的整数 targetSum ,判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。
叶子节点 是指没有子节点的节点。
四、实现
/**
* 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 boolean hasPathSum(TreeNode root, int targetSum) {
if(root == null) return false;
targetSum = targetSum - root.val;
if(targetSum==0 && root.left == null && root.right == null){
return true;
}
return hasPathSum(root.left, targetSum) || hasPathSum(root.right, targetSum);
}
}