路径总和

问题描述：

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

说明: 叶子节点是指没有子节点的节点。

示例: 
给定如下二叉树，以及目标和 sum = 22，

              5
             / 
            4   8
           /   / 
          11  13  4
         /        
        7    2      1
返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/path-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解决方案

修改二叉树深度优先遍历即可，不过需要注意的是叶子结点时没有子节点的结点。实现代码如下：

class Solution {
    public boolean hasPathSum(TreeNode root, int sum) {
        if(root == null){
            return false;
        }
        return dfs(root, sum);
    }
    public boolean dfs(TreeNode root, int sum){
        sum -= root.val;
        if(root.left == null && root.right == null){
            return sum == 0;
        }
        boolean ans = false;
        if(root.left != null){
            ans = ans || dfs(root.left, sum);
        }
        if(root.right != null){
            ans = ans || dfs(root.right, sum);
        }
        return ans;
    }
}

路径总和II

问题描述：

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

说明: 叶子节点是指没有子节点的节点。

示例:
给定如下二叉树，以及目标和 sum = 22，

              5
             / 
            4   8
           /   / 
          11  13  4
         /      / 
        7    2  5   1
返回:

[
   [5,4,11,2],
   [5,8,4,5]
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/path-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解决方案

直接dfs。

class Solution {
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        if(root == null){
            return new ArrayList<>();
        }
        List<List<Integer>> result = new ArrayList<>();
        dfs(root, sum, new Stack<>(), result);
        return result;
    }
    public void dfs(TreeNode root, int sum, Stack<Integer> stack, List<List<Integer>> result){
        sum -= root.val;
        stack.push(root.val);
        if(root.left == null && root.right == null){
            if(sum == 0){
                result.add(new ArrayList<>(stack));
            }
            stack.pop();
            return;
        }
        if(root.left != null){
            dfs(root.left, sum, stack, result);
        }
        if(root.right != null){
            dfs(root.right, sum, stack, result);
        }
        stack.pop();
    }
}