程序员面试一百题-04-在二元树中找出和为某值的所有路径

版权声明：知识无界限，大家可自由转载。 https://blog.csdn.net/xidianbaby/article/details/84927056

1-题目 :
输入一个整数和一棵二元树；从树的根结点开始往下访问一直到叶结点所经过的所有结点形成一条路径；打印出和与输入整数相等的所有路径。

2-示例 :
输入整数22和如下二元树，则应答应出两条路径10-12和10-5-7。

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

4-代码 :

//二元树节点的数据结构定义
struct BinaryTreeNode
{
    int m_nValue;
    BinaryTreeNode *m_pLeft;
    BinaryTreeNode *m_pRight;
};

//找出和为某值的路径
//参数列表 : [二元树的一个结点, 期待的某值, 当前路径, 当前的和]
void FindPath(BinaryTreeNode *pTreeNode, int expectedSum, std::vector<int> &path, int &currentSum)
{
    //若该结点为空,则返回
    if (!pTreeNode)
    {
        return;
    }

    currentSum += pTreeNode->m_nValue;
    path.push_back(pTreeNode->m_nValue);

    //若该结点没有左右子树,则为叶子结点
    bool isLeaf = (!pTreeNode->m_pLeft && !pTreeNode->m_pRight);
    //如果该结点是叶子结点并且当前和为期待和,则输出该路径
    if (currentSum == expectedSum && isLeaf)
    {
        std::vector<int>::iterator iter = path.begin();
        while (iter != path.end())
        {
            std::cout << *iter << " ";
        }
        std::cout << std::endl;
    }

    //如果该结点不是叶子结点,则转到它的孩子
    if (pTreeNode->m_pLeft)
    {
        FindPath(pTreeNode->m_pLeft, expectedSum, path, currentSum);
    }
    if (pTreeNode->m_pRight)
    {
        FindPath(pTreeNode->m_pRight, expectedSum, path, currentSum);
    }

    //当该结点访问结束,返回到它的父结点并,然后路径中删除该结点,最后从当前和中减去该结点的值
    currentSum -= pTreeNode->m_nValue;
    path.pop_back();
}