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

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




猜你喜欢

转载自blog.csdn.net/LonelyGambler/article/details/54236102