算法（44）--二叉树中和为某一值的路径(三)

前言

仅记录学习笔记，如有错误欢迎指正。

题目

给定一个二叉树root和一个整数值 sum ，求该树有多少路径的的节点值之和等于 sum 。 1.该题路径定义不需要从根节点开始，也不需要在叶子节点结束，但是一定是从父亲节点往下到孩子节点 2.总节点数目为n 3.保证最后返回的路径个数在整形范围内

示例

• 输入:{1,2,3,4,5,4,3,#,#,-1},6
• 输出: 3

解法一

dfs

public int FindPath (TreeNode root, int sum) {
        // write code here
        // FindPath 方法定义为从 root 节点开始，必须包括 root 节点，有多少路径的和为 sum
        if (root == null) {
            return 0;
        }
        dfs(root, sum,  0);
        FindPath(root.left, sum);
        FindPath(root.right, sum);
        return ans;
    }
    private int ans;
    public void dfs(TreeNode root, int sum,  int target) {
        if (root == null) {
            return ;
        }
        target += root.val;
        if (target == sum) {
            ans ++;
        }
        dfs(root.left, sum, target);
        dfs(root.right, sum, target);
        target -= root.val;
    }

解法二

换个思路，简单很多

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @param sum int整型 
     * @return int整型
     */
    int res = 0;
    public int FindPath (TreeNode root, int sum) {
        // write code here
   
        if(root == null){
            return res;
        }
        fun(root,sum);
        FindPath(root.left,sum);
        FindPath(root.right,sum);
        return res;
    }
    private void fun(TreeNode root, int sum ){
        
        if(root == null){
            return;
        }
        sum-=root.val;
        if(0 == sum){
            res ++;
        }
         fun (root.left,sum);
         fun (root.right,sum);
        }
    
}