在二元树中找出和为某一值的所有路径

KevinIceFire

于 2013-09-26 19:37:56 发布

阅读量571

点赞数

CC 4.0 BY-SA版权

分类专栏：算法

本文链接：https://blog.youkuaiyun.com/nishiwodeangel/article/details/12066269

算法专栏收录该内容

21 篇文章

订阅专栏

本文介绍了一个算法，该算法接收一个整数及一棵二叉树作为输入，并找出所有从根节点到叶子节点的路径，这些路径上的节点值之和等于给定的整数。文中提供了一个具体的实现代码，通过递归方式遍历二叉树并记录符合条件的路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：输入一个整数和一棵二元树。从树的根结点开始往下访问一直到叶结点所经过的所有结点形成一条路径。打印出和与输入整数相等的所有路径。

struct BinaryTreeNode
{
	int m_nValue;
	BinaryTreeNode *m_pLeft;
	BinaryTreeNode *m_pRight;
};
void FindPath (BinaryTreeNode *pTreeNode, int expectedSum, 
	vector<int> &path, int &currentSum)
{
	if (pTreeNode == NULL)
		return;
	currentSum += pTreeNode->m_nValue;
	path.push_back(pTreeNode->m_nValue);
	bool isLeaf = (!pTreeNode->m_pLeft && !pTreeNode->m_pRight);
	if (currentSum == expectedSum && isLeaf)
	{
		vector<int>::iterator iter = path.begin();
		for (; iter != path.end(); ++iter)
			cout << *iter << '\t';
		cout << endl;
	}
	if (pTreeNode->m_pLeft) 
		FindPath(pTreeNode->m_pLeft, expectedSum, path, currentSum); 
	if (pTreeNode->m_pRight) 
		FindPath(pTreeNode->m_pRight, expectedSum, path, currentSum);
	currentSum -= pTreeNode->m_nValue;
	path.pop_back();
}