题目:输入一棵二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
测试用例:
功能测试(二叉树中有一条、多条符合条件的路径,二叉树中没有符合条件的路径)
特殊输入测试(指向二叉树根结点的指针为NULL指针)
代码如下:
struct BinaryTreeNode
{
int m_nValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_pRight;
}
void FindPathTree(BinaryTreeNode* pRoot,int expectSum)
{
if(pRoot==NULL)
return;
std::vector<int> path;
int currentSum=0;
FindPath(pRoot,expectSum,path.currentSum);
}
void FindPath(BinaryTreeNode* pRoot,int expectSum,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==expectSum && 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, expectSum, path, currentSum);
if(pRoot->m_pRight!=NULL)
FindPath(pRoot->m_pRight, expectSum, path, currentSum);
//在返回到父结点之前,在路径上删除当前结点
path.pop_back();
}