码迷,mamicode.com
首页 > 其他好文 > 详细

五:二叉树中和为某一直的路径

时间:2015-07-07 22:52:55      阅读:176      评论:0      收藏:0      [点我收藏+]

标签:二叉树中和为某一直的路径   路径和   

例如:求和为22的路径

技术分享


求值步骤

技术分享


规律:当用前序遍历的方式访问到某一节点时,我们把这个节点添加到路径上,并累加该节点的值,如果该节点为叶子节点并且路径中节点值的和刚好等于输入的整数,则当前的路径符合要求,我们把它打印出来。如果当前节点不是叶节点,则继续访问它的子节点。当前节点访问结束后,递归函数将自动回到它的父节点。因此我们在函数退出之前要在路径上删除当前节点,并减去当前节点的值,以确保返回父节点时路径刚好是从根节点到父节点的路径。不难看出保存路径的数据结构实际上是一个栈,因为路径要与递归调用状态一致,而递归调用的本质就是一个压栈和出栈的过程。

但是由于使用栈不便于路径的输出,所以可以借助于vector的push_back和pop_back在尾部增删路径节点


代码如下:

voidFindPath

(

    BinaryTreeNode*   pRoot,       

    int               expectedSum, 

    std::vector<int>& path,        

    int&              currentSum

)

{

    currentSum += pRoot->m_nValue;

    path.push_back(pRoot->m_nValue);

 

    // 如果是叶结点,并且路径上结点的和等于输入的值

    // 打印出这条路径

    bool isLeaf = pRoot->m_pLeft == NULL&& pRoot->m_pRight == NULL;

    if(currentSum == expectedSum &&isLeaf)

    {

        printf("A path is found: ");

       std::vector<int>::iterator iter = path.begin();

        for(; iter != path.end(); ++ iter)

            printf("%d\t", *iter);

       

        printf("\n");

    }

 

    // 如果不是叶结点,则遍历它的子结点

    if(pRoot->m_pLeft != NULL)

        FindPath(pRoot->m_pLeft,expectedSum, path, currentSum);

    if(pRoot->m_pRight != NULL)

        FindPath(pRoot->m_pRight,expectedSum, path, currentSum);

 

    // 在返回到父结点之前,在路径上删除当前结点,

    // 并在currentSum中减去当前结点的值

    currentSum -= pRoot->m_nValue;

    path.pop_back();

}

 

voidCallFindPath(BinaryTreeNode* pRoot, int expectedSum)

{

    if(pRoot == NULL)

        return;

 

    std::vector<int> path;

    int currentSum = 0;

    FindPath(pRoot, expectedSum, path,currentSum);

}



版权声明:本文为博主原创文章,未经博主允许不得转载。

五:二叉树中和为某一直的路径

标签:二叉树中和为某一直的路径   路径和   

原文地址:http://blog.csdn.net/day__day__up/article/details/46793741

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!