经典算法学习——求二叉树节点和为sum的路径

算法设计 同时被 2 个专栏收录
79 篇文章 1 订阅
80 篇文章 78 订阅

       在一棵二叉树中,有多少个叶子节点,就会有多少条从根节点到叶子节点的路径。如果给每一个节点一个data值,那么每一条路径经过的节点data加起来就为sum,现在根据给定的sum,请找出有哪些路径符合这个sum。代码上传至 https://github.com/chenyufeng1991/BinaryTreePath  。

       整体思路就是递归的去检索,每当经过一个节点的时候,就把节点值放入vector中,当到达叶子节点时,判断此时vector中的和是否等于给定sum,等于的话表示这是一条符合条件的路径,打印该路径。否则清空vector,继续检索。

核心代码如下:

void PathLength(Node *node, vector<int> &vec, int sum)
{
    if (node == NULL)
    {
        return;
    }

    if (node ->lChild != NULL)
    {
        vec.push_back(node->data);
        PathLength(node->lChild,vec,sum);
    }

    if (node ->rChild != NULL)
    {
        vec.push_back(node->data);
        PathLength(node->rChild,vec,sum);
    }

    if (node ->lChild == NULL || node ->rChild == NULL)
    {
        vec.push_back(node->data);
        int length = 0;
        for (int i = 0; i < vec.size(); i++)
        {
            length += vec[i];
        }

        if (length == sum)
        {
            for (int i = 0; i < vec.size(); i++)
            {
                cout << vec[i] << " ";
            }
            cout << endl;
        }
        vec.clear();
    }
}


  • 0
    点赞
  • 0
    评论
  • 1
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值