leetcode-树-从根到叶的二进制数之和

最新推荐文章于 2022-05-30 08:03:41 发布

FF的CC

最新推荐文章于 2022-05-30 08:03:41 发布

阅读量119

点赞数

分类专栏： LeetCode 数据结构与算法文章标签： leetcode 算法二叉树 dfs

本文链接：https://blog.csdn.net/cix1314/article/details/110410977

版权

数据结构与算法同时被 2 个专栏收录

15 篇文章 0 订阅

订阅专栏

LeetCode

14 篇文章 0 订阅

订阅专栏

该博客主要介绍了如何解决一个关于二叉树的问题，即计算从根节点到每个叶子节点的二进制路径所表示的数字之和。通过前序遍历的方法，将每条路径转换为二进制数并累加，最终得到答案。示例展示了不同输入树结构下的输出结果，并提供了相应的代码实现。

摘要由CSDN通过智能技术生成

给出一棵二叉树，其上每个结点的值都是 0 或 1 。每一条从根到叶的路径都代表一个从最高有效位开始的二进制数。例如，如果路径为 0 -> 1 -> 1 -> 0 -> 1，那么它表示二进制数 01101，也就是 13 。

对树上的每一片叶子，我们都要找出从根到该叶子的路径所表示的数字。

返回这些数字之和。题目数据保证答案是一个 32 位整数。

示例 1：

输入：root = [1,0,1,0,1,0,1]
输出：22
解释：(100) + (101) + (110) + (111) = 4 + 5 + 6 + 7 = 22
示例 2：

输入：root = [0]
输出：0
示例 3：

输入：root = [1]
输出：1
示例 4：

输入：root = [1,1]
输出：3

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sum-of-root-to-leaf-binary-numbers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

二叉树的前序遍历

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int sumRootToLeaf(TreeNode* root) {
        int res=0;
        int curSum=0;
        SumRootToLeaf(root,curSum,res);
        return res;
    }

    void SumRootToLeaf(TreeNode* node, int curSum, int& res)
    {
        if(node==nullptr)
        {
            return;
        }

        if(node->left==nullptr && node->right==nullptr)
        {
            curSum = curSum*2+node->val;
            res += curSum;
            return;
        }

        curSum = curSum*2 + node->val;
        SumRootToLeaf(node->left,  curSum, res);
        SumRootToLeaf(node->right, curSum, res);
    }
};