leetcode 112. 路径总和

112 路径总和

给你二叉树的根节点 root 和一个表示目标和的整数 targetSum ，判断该树中是否存在 根节点到叶子节点 的路径，这条路径上所有节点值相加等于目标和 targetSum 。

叶子节点 是指没有子节点的节点。

示例 1：

输入：root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
输出：true

示例 2：

输入：root = [1,2,3], targetSum = 5
输出：false

示例 3：

输入：root = [1,2], targetSum = 0
输出：false

/**
 * 递归方式
 */
public static boolean hasPathSum(TreeNode root, int sum) {
    if (root == null) {
        return false;
    }

    // 如果是叶子节点，判断节点的值与sum是否相等
    if (root.left == null && root.right == null) return root.val == sum;

    return hasPathSum(root.left, sum - root.val)
            || hasPathSum(root.right, sum - root.val);
}

113 路径总和II

给你二叉树的根节点 root 和一个整数目标和 targetSum ，找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。

叶子节点 是指没有子节点的节点。

 

示例 1：

输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出：[[5,4,11,2],[5,8,4,5]]

示例 2：

输入：root = [1,2,3], targetSum = 5
输出：[]

示例 3：

输入：root = [1,2], targetSum = 0
输出：[]

List<List<Integer>> ret = null;
Deque<Integer> path = null

public List<List<Integer>> pathSum(TreeNode root, int sum) {
    ret =  new LinkedList<List<Integer>>();
    path = new LinkedList<Integer>();
    dfs(root, sum);
    return ret;
}

public void dfs(TreeNode root, int sum) {
    if (root == null) {
        return;
    }
    path.offerLast(root.val);
    sum -= root.val;
    if (root.left == null && root.right == null && sum == 0) {
        ret.add(new LinkedList<Integer>(path));
    }
    dfs(root.left, sum);
    dfs(root.right, sum);
    path.pollLast();
}

437 路径总和 III

给定一个二叉树的根节点 root ，和一个整数 targetSum ，求该二叉树里节点值之和等于 targetSum 的 路径 的数目。

路径 不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

示例 1：

输入：root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
输出：3
解释：和等于 8 的路径有 3 条，如图所示。

示例 2：

输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出：3

深度优先+递归

/**
 * 深度优先 ：递归
 * @param root
 * @param targetSum
 * @return
 */
public static int pathSum(TreeNode root, int targetSum) {
    if (root == null) return 0;

    // 从当前节点开始遍历,判断是否有一条符合条件的路径
    int sum = findPath(root, 0, targetSum);
    // 分别从当前节点的左右子树开始遍历
    if (root.left != null) sum += pathSum(root.left, targetSum);
    if (root.right != null) sum += pathSum(root.right, targetSum);

    return sum;
}

/**
 * 从某个节点开始判断是否存在到某个节点的路径和等于sum
 *
 * @param root
 * @param pre
 * @param sum
 * @return
 */
public static int findPath(TreeNode root, int pre, int sum) {
    if (root == null) return 0;

    int curSum = root.val + pre;
    int cnt = curSum == sum ? 1 : 0;

    if (root.left != null) cnt += findPath(root.left, curSum, sum);
    if (root.right != null) cnt += findPath(root.right, curSum, sum);
    return cnt;
}