这个题目需要将非递归中序遍历中的入栈方法稍微改进一下
1、对任意节点Cur若不为空入栈,并将其左孩子置为Cur,直到Cur为空;
2、判断栈是否为空,若不为:将栈顶元素赋给Cur(但并不出栈),若Cur的右孩子为空或Pre与Cur的右孩子值相等,则判断此时栈中所有元素的和是否与特定的sum值相等,并将栈顶元素Pop给Pre;将Cur的右孩子置为Cur;
3、直到Cur为空且栈为空时,遍历结束
ElementType SumOfStack(Stack St)
{
int sum, i;
for (i=0,sum=0; i<St->count; i++)
{
sum += St->Elment[i]->Element;
}
return sum;
}
void PrtStack(Stack St)
{
int i;
for (i=0; i<St->count; i++)
{
printf("%3d",St->Elment[i]->Element);
}
}
void PrtSumPath(SearchTree T)
{
Stack St;
SearchTree Cur, Pre = NULL;
int sum;
St = NewStack();
Cur = T;
scanf_s("%d",&sum);
while(Cur || !IsEmpty(St))
{
while(NULL != Cur)
{
Push(St, Cur);
Cur = Cur->Left;
}
if (!IsEmpty(St))
{
Cur = Top(St);
if (NULL == Cur->Right || Pre == Cur->Right)
{
if (sum == SumOfStack(St))
{
PrtStack(St);
}
Pre = Pop(St);
}
Cur = Cur->Right;
}
}
}