100题_04 在二元树中找出和为某一值的所有路径
题目:输入一个整数和一棵二元树。从树的根结点开始往下访问一直到叶结点所经过的所有结点形成一条路径。打印出和与输入整数相等的所有路径。
例如输入整数22和如下二元树
例如输入整数22和如下二元树
10
/
5 12
/
4 7
则打印出两条路径:10, 12和10, 5, 7。
想想其实把所有的路径找出来,然后求和去算就可以解决。但是针对这题,有些路径可以在完全找出来之前就否定了……所以需要剪枝。
利用递归的思想:对于结点10,我们只要在其左右两个子树中找出路径为12的就可了,所以很快就可以得到结果,注意,我们需要记录路径。
下面是代码:
View Code
void BinaryTree::FindPath(BinaryTreeNode *root, int va, vector<int> &path)
{
if (root == NULL)
return;
path.push_back(root->m_nValue);
int value = va - root->m_nValue;
if (value == 0)
{
if (root->m_pLeft == NULL || root->m_pRight == 0)
{
for (vector<int>::iterator it = path.begin(); it != path.end(); it++)
cout<<*it<<' ';
cout<<endl;
}
path.pop_back();
return;
}
if (value < 0)
return;
if (root->m_pLeft)
FindPath(root->m_pLeft, value, path);
if (root->m_pRight)
FindPath(root->m_pRight, value, path);
path.pop_back();
}
{
if (root == NULL)
return;
path.push_back(root->m_nValue);
int value = va - root->m_nValue;
if (value == 0)
{
if (root->m_pLeft == NULL || root->m_pRight == 0)
{
for (vector<int>::iterator it = path.begin(); it != path.end(); it++)
cout<<*it<<' ';
cout<<endl;
}
path.pop_back();
return;
}
if (value < 0)
return;
if (root->m_pLeft)
FindPath(root->m_pLeft, value, path);
if (root->m_pRight)
FindPath(root->m_pRight, value, path);
path.pop_back();
}
本文基于署名 2.5 中国大陆许可协议发布,欢迎转载,演绎或用于商业目的,但是必须保留本文的署名小橋流水(包含链接)。如您有任何疑问或者授权方面的协商,请给我发邮件。