LCR 049. 求根节点到叶节点数字之和 - 力扣(LeetCode)
理解题意分析子问题:给一个头节点,要返回该头结点左右子树的根结点到叶节点数字和。此处还需注意:在获取根结点到叶节点数字和的时候,要传递一个参数:从根结点到当前节点的数字和,比如,当前节点为 9,此处要传递根结点到当前节点的数字和,也就是 4;
函数头 -> int dfs(TreeNode root,int presum);
分析子问题:
1. 根据拿到的 presum 参数,形成新的和:presum*10 + 当前节点的值;
2. 让左子树继续遍历,接受返回值;
3. 让右子树继续遍历,接受叠加返回值;
4. 把叠加后的返回值返回会上一级;
思考递归出口:当节点的左右子树都为null的时候,进行返回,但返回的实际是要在加完该叶子节点后,也就是上述分析子问题中的第一步,要走完第一步才可以进行返回;
代码实现
class Solution {
public int sumNumbers(TreeNode root) {
return dfs(root,0); // 头节点的前置和为0
}
public int dfs(TreeNode root,int presum){
presum = presum*10 + root.val;
if(root.left == null && root.right == null) return presum; // 递归出口
int ret = 0; // ret 来接收左右子树的返回值之和
if(root.left != null) ret += dfs(root.left,presum);
if(root.right != null) ret += dfs(root.right,presum);
return ret;
}
}