DFS Sum Root to Leaf Numbers

最新推荐文章于 2024-01-17 15:33:38 发布

qwurey

最新推荐文章于 2024-01-17 15:33:38 发布

阅读量385

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/yeruby/article/details/45822355

leetcode 专栏收录该内容

234 篇文章

订阅专栏

本文介绍了一种使用递归方法求和二叉树中所有节点数值的算法，优化了空间复杂度并减少了中间变量的使用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路：

DFS。

时间复杂度O(N)，空间复杂度O(logN)。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
private:
    void dfs(int sum, int &total_sum, TreeNode* root) {
        if(root == NULL) {
            return;
        }
        if(root->left == NULL && root->right == NULL) {
            sum = sum*10 + root->val;
            total_sum += sum;
            return;
        }
        dfs(sum*10+root->val, total_sum, root->left);
        dfs(sum*10+root->val, total_sum, root->right);
    }
public:
    int sumNumbers(TreeNode* root) {
        int sum = 0;
        int total_sum = 0;
        dfs(sum, total_sum, root);
        return total_sum;
    }
};

优化，去掉sum和total_sum中间变量。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
private:
    int dfs(TreeNode* root, int sum) {
        if(root == NULL) return 0;
        if(root->left == NULL && root->right == NULL) {
            return 10*sum + root->val;
        }
        return dfs(root->left, 10*sum + root->val) + dfs(root->right, 10*sum + root->val);
    }
public:
    int sumNumbers(TreeNode* root) {
        return dfs(root, 0);
    }
};