前言
仅记录学习笔记,如有错误欢迎指正。
题目
给定一个二叉树root和一个整数值 sum ,求该树有多少路径的的节点值之和等于 sum 。 1.该题路径定义不需要从根节点开始,也不需要在叶子节点结束,但是一定是从父亲节点往下到孩子节点 2.总节点数目为n 3.保证最后返回的路径个数在整形范围内示例
- 输入:{1,2,3,4,5,4,3,#,#,-1},6
- 输出: 3
解法一
dfspublic 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);
}
}