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

最新推荐文章于 2017-09-05 18:32:05 发布

shiquxinkong

最新推荐文章于 2017-09-05 18:32:05 发布

阅读量919

点赞数 1

分类专栏：基础算法文章标签：遍历递归

本文链接：https://blog.csdn.net/shiquxinkong/article/details/22820249

版权

基础算法专栏收录该内容

52 篇文章 0 订阅

订阅专栏

Q：输入一颗二叉树和一个整数，打印出二叉树中结点和为输入整数的所有路径。从树的根结点开始到叶子节点所经过的结点形成一条路径。结点定义如下：

struct BinaryTreeNode
{
    int    m_nValue;
    BinaryTreeNode *m_pLeft;
    BinaryTreeNode *m_pRight;
}

这个问题直白点就是二叉树的遍历，在遍历的过程中要随时记录一些数据，在数据满足一定条件时，做一些特定的动作。

我们很自然的想到二叉树的三种遍历，以先根(前序)遍历来说，其遍历的递归模板是：

void PreOrder(BinaryTreeNode *root)
{
	if(!root) return;
	visit(root);
	PreOrder(root->m_pLeft);
	PreOrder(root->m_pRight);
}

真对不同的要求visit()可以做不同的处理，这里的要求是判断当前的结点和是否等于给定的值，同时形成了一条路径(为叶子结点)，若是，则输出这条路径。

void FindPath(BinaryTreeNode *pRoot, int expecteSum, int curSum)
{
    if(!pRoot) return;
	curSum += pRoot->m_nValue;
	path.push_back(pRoot);
	if(curSum == expecteSum && IsLeaf(pRoot))
	{
		std::vector<BinaryTreeNode *>::iterator iter = path.begin();
		while (iter != path.end())
		{
			std::cout << (*iter)->m_nValue << " ";
			++iter;
		}
		std::cout << std::endl;
	}
	if(pRoot->m_pLeft)
	FindPath(pRoot->m_pLeft,expecteSum,curSum);
	if(pRoot->m_pRight)
	FindPath(pRoot->m_pRight,expecteSum,curSum);
	path.pop_back();
}