Q25:二叉树中和替某一值的路径

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

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();
}