剑指offer之面试题34：二叉树中和为某一值的路径

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

1、题目

输入一颗二叉树和一个整数，打印出二叉树中节点值的和为输入整数的所有路径。

从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。二叉树节点的定义如下：

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

输入参数：一颗二叉树、一个整数

输出结果：二叉树的路径，或者为空

2、解题

本题需要把握两个重点：

• 二叉树直到走到叶子节点才算一条路径，因此最终的比较在叶子节点
• 二叉树的节点没有指向父节点的指针，无法知道前面经过了哪些节点，需要一个栈来保存经过的节点的值

解题的完整步骤为：

• 主函数
  • 鲁棒性检查
  • 栈、当前总和值的初始化
• 递归函数
  • 根节点入栈、总和值增加
  • 判断当前节点是否到达叶节点（已经形成一条完整路径）且当前总和值与期待值相等
    • 若相等，打印栈中的完整数据
  • 如果不是叶节点，则遍历其子节点
  • 在返回父节点之前，在路径上删除该节点（元素出栈）

3、完整代码

void FindPath(BinaryTreeNode* pRoot, int expectedSum) {
	//1、鲁棒性检查
	if (pRoot == nullptr)
		return;
	//2、初始化
	std::vector<int> path;
	int currentSum = 0;
	FindPath(pRoot, expectedSum, path, currentSum);
}

void FindPath(BinaryTreeNode* pRoot, int expectedSum, std::vector<int>& path, int currentSum) {
	//3、首先将根节点入栈，并将值加入总和
	currentSum += pRoot->m_nValue;
	path.push_back(pRoot->m_nValue);
	//4、判断是否到叶节点且总和与期待值相等
    bool isLeaf = (pRoot->m_pLeft == nullptr && pRoot->m_pRight == nullptr);
	if (currentSum == expectedSum && isLeaf) {
		printf("A path is found:");
		std::vector<int>::iterator iter = path.begin();
		for (; iter != path.end(); iter++)
			printf("%d\t", *iter);
		printf("\n");
	}
	//5、未到达叶节点，遍历它的子节点
	if (pRoot->m_pLeft != nullptr)
		FindPath(pRoot->m_pLeft, expectedSum, path, currentSum);
	if (pRoot->m_pRight != nullptr)
		FindPath(pRoot->m_pRight, expectedSum, path, currentSum);
	//6、返回父节点时，将该节点从路径上删除
	path.pop_back();
}

4、注意点

• 遍历vector容器时迭代器的使用
• 鲁棒性检查不需要考虑输入的整数
• 递归子节点后，记得将该节点出栈
• 时刻注意一条完整路径需要到达叶子结点才算