1.题目
给定一个二叉树root和一个整数值 sum ,求该树有多少路径的的节点值之和等于 sum 。
1.该题路径定义不需要从根节点开始,也不需要在叶子节点结束,但是一定是从父亲节点往下到孩子节点
2.总节点数目为n
3.保证最后返回的路径个数在整形范围内(即路径个数小于231-1)
2.解法
2.1 树的结构
class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
2.2 解法1
这道题的答案可能是树路径中的那一截,并没有要求达到叶子节点。且只需要总路径条数。
Map<TreeNode,Integer> counter = new HashMap<>();
public int FindPath(TreeNode root, int sum) {
// write code here
if(root==null) return 0;
int left = FindPath(root.left,sum);
int right = FindPath(root.right,sum);
int centre = helper(root,root,sum);
return left+right+centre;
}
public int helper(TreeNode root,TreeNode cur,int sum){
if(root==null||cur==null) return 0;
int n = 0;
if(cur.val==sum) n = 1;
return n+helper(root,cur.left,sum-cur.val)+helper(root,cur.right,sum-cur.val);
总结
FindPath利用helper函数寻路,其本身也递归调用自己FindPath 。
算法系列在github上有一个开源项目,主要是本系列博客的demo代码。https://github.com/forestnlp/alg
如果您对软件开发、机器学习、深度学习有兴趣请关注本博客,将持续推出Java、软件架构、深度学习相关专栏。
您的支持是对我最大的鼓励。