宏观角度认识递归之求根节点到叶节点数字之和

LCR 049. 求根节点到叶节点数字之和 - 力扣(LeetCode)

理解题意分析子问题:给一个头节点,要返回该头结点左右子树的根结点到叶节点数字和。此处还需注意:在获取根结点到叶节点数字和的时候,要传递一个参数:从根结点到当前节点的数字和,比如,当前节点为 9,此处要传递根结点到当前节点的数字和,也就是 4;

宏观角度认识递归之求根节点到叶节点数字之和_第1张图片

函数头 -> int dfs(TreeNode root,int presum);

分析子问题:

1. 根据拿到的 presum 参数,形成新的和:presum*10 + 当前节点的值;

2. 让左子树继续遍历,接受返回值;

3. 让右子树继续遍历,接受叠加返回值; 

4. 把叠加后的返回值返回会上一级;

宏观角度认识递归之求根节点到叶节点数字之和_第2张图片

思考递归出口:当节点的左右子树都为null的时候,进行返回,但返回的实际是要在加完该叶子节点后,也就是上述分析子问题中的第一步,要走完第一步才可以进行返回;

宏观角度认识递归之求根节点到叶节点数字之和_第3张图片

代码实现 

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;
    }
}

 

你可能感兴趣的:(递归,算法,数据结构,递归)