题目:
输入一颗二叉树和一个整数,打印出二叉树节点中节点值的和为输入整数的所有路径 。从树的根节点往下所经过的节点的形成的路径。
思路:
先序遍历的方式访问某一个节点时,把该节点添加到路径上,并且累加节点中的值。如果当前节点时叶节点并且累加的值等于输入的值,那么我们就把这条路径打印出来。如果不是叶节点,那么就访问当前节点的左右子树。当前节点的左右子树都访问完了,那么久返回到父节点,(在返回父节点时,一定要把当前节点删除当前路径)。
code:
<span style="font-size:14px;">void FindPath(node* pRoot, int expection)
{
if (pRoot == NULL)
return ;
std::vector<int> path;
int currentSum = 0;
FindPath(pRoot, expection, path, currentSum);
}
void FindPath(node *pRoot, int expection, std::vector<int>& path, int currentSum)
{
currentSum += pRoot->data;
path.push_back(pRoot->data);
bool IsLeef = pRoot->pLeft == NULL && pRoot->pRight == NULL;
if (currentSum == expection && IsLeef)
{
printf("find path : \n");
std::vector<int> iterator it = path.begin();
for (; it != path.end(); ++it)
{
cout << *it << "->";
}
printf("\n");
}
if (pRoot->pLeft)
{
FindPath(pRoot->pLeft, expection, path, currentSum);
}
if (pRoot->pRight)
{
FindPath(pRoot->pRight, expection, path, currentSum);
}
//返回父节点之前,删除路径当前节点
path.pop_back();
}</span>