在二叉树中打印出从某个节点(r)开始和为定值(sum)的所有路径

来源:互联网 发布:qq邮箱mac电脑版下载 编辑:程序博客网 时间:2024/04/28 03:53

通过在函数中定义两个静态栈,对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();}




0 0
原创粉丝点击