通过在函数中定义两个静态栈,对r的每个孩子进行遍历和累计,当遍历到的孩子的节点和与sum相等时就打印出这条路径。
至于遍历的方法,个人认为递归可以比较简单的实现。
本文为模板二叉树中的一个子函数,如果想要整个模板二叉树类的代码请联系本人。
不多说,直接上代码吧,让注释来慢慢说话。
template <class T>
void tree<T>::print_route(tree_node<T> *r,T sum)
{
static stack<tree_node<T> > get_s,put_s;
if(r==NULL)
return;
sum-=r->get_data();
get_s.push(r);//遍历树的节点的同时用get_s保存节点的存放的数据
if(sum<=0)
{
if(0==sum)
{
while(!get_s.empty())
put_s.push(get_s.pop());//将get_s中的节点倒入put_s中,这样保证打印的路径从r开始
while(!put_s.empty())
{
cout<<put_s.get_top()->get_data()<<" ";
get_s.push(put_s.get_top());
put_s.pop();
}
cout<<endl;
}
sum+=get_s.pop()->get_data();
return;
}
print_route(r->get_left(),sum);//递归进入r的左子树
print_route(r->get_right(),sum);//递归进入r的右子树
sum+=get_s.pop()->get_data();
}