二叉树中和为某一值的路径

最新推荐文章于 2022-11-09 20:08:03 发布

Kaycsdner

最新推荐文章于 2022-11-09 20:08:03 发布

阅读量491

点赞数

分类专栏：常见数据结构与算法面试题文章标签： struct c++ 二叉树遍历

本文链接：https://blog.csdn.net/petersmart123/article/details/49680661

版权

常见数据结构与算法面试题专栏收录该内容

16 篇文章 0 订阅

订阅专栏

题目：输入一颗二叉树和一个整数，打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始一直到叶节点所经过的节点形成的一条路径。

解答：采用类似于前序遍历的过程，依次累加节点值，直到到达叶子节点并且累加和与期望值相等，就打印路径。

struct TreeNode{
	int m_data;
	TreeNode* m_pLeft;
	TreeNode* m_pRight;
	TreeNode(const int data = 0,TreeNode* left=NULL,TreeNode* right=NULL):
		m_data(data),m_pLeft(left),m_pRight(right){}
};

bool IsLeaf(TreeNode* pNode)
{
	if(NULL == pNode)
		return false;
	return (NULL==pNode->m_pLeft && NULL==pNode->m_pRight)?true:false;
}
void SumOfPath(TreeNode* pRoot,vector<int>& vec,int expectedSum,int sum)
{
	sum += pRoot->m_data;
	vec.push_back(pRoot->m_data);

	if(IsLeaf(pRoot) && sum == expectedSum)
	{
		for(int index = 0; index < (int)vec.size(); ++index)
			cout<<vec[index]<<' ';
		cout<<endl;
	}

	if(pRoot->m_pLeft)
		SumOfPath(pRoot->m_pLeft,vec,expectedSum,sum);
	if(pRoot->m_pRight)
		SumOfPath(pRoot->m_pRight,vec,expectedSum,sum);
	vec.pop_back();
}

void FindPath(TreeNode* pRoot,int expectedSum)
{
	vector<int> vec;
	int sum = 0;
	SumOfPath(pRoot,vec,expectedSum,sum);
}