题目描述:给定一棵二叉树和二叉树中一个节点,输出根节点到指定节点间的路径。
10
/ \
5 12
/ \
4 7
指定节点7,那么输出路径应该是10-5-7。
分析与解法:
这个题目是在我做过蛮多二叉树的题目之后总结的一道题目。发现很多题目都可以抽象出来这个题目。其实二叉树本来就是最典型的递归数据结构,只要完全掌握三种遍历方法,一切题目都是浮云。
这道题目和博客http://blog.csdn.net/getnextwindow/article/details/23326843输出所有满足条件的路径颇为相似。只不过这道题目只要找到目的路径就要尽快返回,不要再遍历下去。不多解释,我会在代码中给予注释。
代码如下:
bool specialPath(Node *pRoot,Node *pNode,vector<int> &v)
{
if(pRoot==NULL)
{
return false;
}
v.push_back(pRoot->m_value);
bool found=false;
if(pRoot==pNode)//还是比较指针稳妥,节点值有可能重复
{
for(int i=0;i<v.size();i++)
cout<<v[i]<<" ";
cout<<endl;
return true;
}
if(!found && pRoot->m_pLeft)
{
found=specialPath(pRoot->m_pLeft,pNode,v);
}
//一旦左子树中找到节点,就不需要再遍历右子树
if(!found && pRoot->m_pRight)
{
found=specialPath(pRoot->m_pRight,pNode,v);
}
if(!found)
v.pop_back();
return found;
}
注:
if(!found) v.pop_back(),只有在左右子树都没发现目标节点才弹出。如果我们仅仅在上面的函数中输出路径,那么没有这条判定语句也是对的,但是如果想使用v保存路径并在函数之外使用,必须加上这条语句,因为没有判定语句,函数回溯过程会弹出v中节点。希望关注和http://blog.csdn.net/getnextwindow/article/details/23326843的区别,上面链接的这篇要求所有满足条件的路径,所以只要知道满足的就打印,然后回退过程要弹出v节点,和本题刚好相反。