面试题25:二叉树中和为某一值的路径
题目:输入一棵二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。从树的根结点开始往下一直到叶子结点所经过的结点形成一条路径。二叉树结点的定义如下:
struct BInaryTreeNode
{
int value;
BinaryTreeNode* pLeft;
BinaryTreeNode* pRight;
}
分析:用先序遍历的方式遍历二叉树,初始时路径为空。访问一个结点时若不到叶子结点且结点的和小于这个值那么就把这个结点添加到路径上并且遍历其子节点,若是叶子结点或值大于这个值,则返回到父结点。若等于这个值则打印。
void FindPath(BinaryTreeNode* pRoot,int expectedSum)
{
if(pRoot == NULL)
{
return ;
}
std::vector<int> path;
int cuurentSum = 0;
FindPath(pRoot,expectedSum,path,currentSum);
}
void FindPath
(
BinaryTreeNode* pRoot,
int expectedSum,
std::vector<int>& path,
int currentSum
)
{
currentSum += pRoot->value;
path.push_back(pRoot->value);
bool isLeaf = pRoot->pLeft == NULL && pRoot->pRight == NULL;
if(current == expectedSum && isLeaf)
{
printf("The path is found");
std::vector<int>::iterator iter = path.begin();
for(;iter != path.end();++iter)
{
printf("%d\t",*iter);
}
printf("\n");
}
if(pRoot->pLeft != NULL)
{
FindPath(pRoot->pLeft,expectedSum,path,currentSum);
}
if(pRoot->Right != NULL)
{
FindPath(pRoot->pRight,expectedSum,path,currentSum);
}
path.pop_back();
}
感觉只要涉及到树的题目,如果不做特殊的要求基本上都会采用递归的写法。