题目:输入一个二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径,从树的根节点开始往下一直到叶节点所经过的结点形成一条路径。
代码:
//和为某一值的路径
//传的是栈的值,所以在函数里面不会改变栈内的值
void FindPath(BTNode* root, int num, Stack s1, int count)
{
assert(root);
StackPush(&s1, root);
BTNode* tmp = StackTop(&s1);
count += tmp->_data;
//和为要求得值,并且该节点的左右子树为空,则将该路径进行输出
if (count == num && root->_left == NULL && root->_right == NULL)
{
while (StackEmpty(&s1))
{
printf("%d <- ", StackTop(&s1)->_data);
StackPop(&s1);
}
printf("root");
printf("\n");
}
if (root->_left != NULL)
{
FindPath(root->_left, num, s1,count);
}
if (root->_right != NULL)
{
FindPath(root->_right, num, s1,count);
}
}
//创建一个栈,存放每次的结点
void Path(BTNode* root, int num)
{
assert(root);
Stack s1, s2;
int count = 0;
StackInit(&s1);
FindPath(root, num, s1,count);
}