题目:输入一个整数和一棵二元树。从树的根结点开始往下访问一直到叶结点所经过的所有结点形成一条路径。打印出和与输入整数相等的所有路径。
例如输入整数22和如下二元树
10
/ \
5 12
/ \
4 7
则打印出两条路径:10, 12和10, 5, 7。
分析:这是百度的一道笔试题,考查对树这种基本数据结构以及递归函数的理解。
当访问到某一结点时,把该结点添加到路径上,并累加当前结点的值。如果当前结点为叶结点并且当前路径的和刚好等于输入的整数,则当前的路径符合要求,我们把它打印出来。如果当前结点不是叶结点,则继续访问它的子结点。当前结点访问结束后,递归函数将自动回到父结点。因此我们在函数退出之前要在路径上删除当前结点并减去当前结点的值,以确保返回父结点时路径刚好是根结点到父结点的路径。我们不难看出保存路径的数据结构实际上是一个栈结构,因为路径要与递归调用状态一致,而递归调用本质就是一个压栈和出栈的过程。
参考代码:
#include <iostream>
using namespace std;
#include <deque>
using std::deque;
template<typename T>
struct Node
{
public:
Node(T value);
T getValue ();
Node *getLeftChld ();
Node *getRightChild ();
void setLeftChild (Node *leftChild);
void setRightChild (Node *rightChild);
private:
T mValue;
Node *mLeftChild;
Node *mRightChild;
};
template<typename T>
Node<T>::Node (T value):mValue(value),mLeftChild(0),mRightChild(0)
{
}
template<typename T>
T
Node<T>::getValue()
{
return mValue;
}
template<typename T>
Node<T>*
Node<T>::getLeftChld()
{
return mLeftChild;
}
template<typename T>
Node<T>*
Node<T>::getRightChild()
{
return mRightChild;
}
template<typename T>
void
Node<T>::setLeftChild(Node *leftChild)
{
mLeftChild = leftChild;
}
template<typename T>
void
Node<T>::setRightChild(Node *rightChild)
{
mRightChild = rightChild;
}
Node<int> *
buildTree()
{
Node<int> *r10 = new Node<int>(10);
Node<int> *r4 = new Node<int>(4);
Node<int> *r5 = new Node<int>(5);
Node<int> *r7 = new Node<int>(7);
Node<int> *r12 = new Node<int>(12);
r10->setLeftChild(r5);
r10->setRightChild(r12);
r5->setLeftChild(r4);
r5->setRightChild(r7);
return r10;
}
template<typename T>
void
getTheValue(Node<T> *node,T expectValue, T & currentValue,deque<T> &path)
{
T nodeValue = node->getValue();
currentValue += nodeValue;
path.push_back(node->getValue());
bool isLeaf = (0 == node->getLeftChld()) && (0 == node->getRightChild());
if ( (currentValue == expectValue) && isLeaf)
{
for ( int i = 0 ; i < path.size(); i ++ )
{
cout << path[i]<<" ";
}
cout << endl;
}
if ( node ->getLeftChld() )
{
getTheValue(node->getLeftChld(),expectValue,currentValue,path);
}
if ( node ->getRightChild() )
{
getTheValue(node->getRightChild(),expectValue,currentValue,path);
}
path.pop_back();
currentValue -= nodeValue;
}
int main(int argc,char *argv[])
{
Node<int> *result = buildTree();
int currentValue = 0;
deque<int> path;
getTheValue<int>(result,22,currentValue,path);
return 0;
}