面试题34: 二叉树中和为某一值的路径

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/love__live1/article/details/82966212

Github

/*******************************************************************
*《剑指Offer——名企面试官精讲典型编程题》C++代码
*
* htfeng
* 2018.10.08
*
* 面试题34: 二叉树中和为某一值的路径
* 题目:输入一颗二叉树和一个整数,打印出二叉树中节点值得和为输入整数得所有路径。
*
* 分析: 对于二叉搜索树的后续遍历,最后一个节点是个分界点,前面的一组
* 数字比它小,后面几个数字比它大,然后前面一组和后面一组也是同样,利用递归。
*******************************************************************/
#include<iostream>
#include<vector>

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

class Solution {
public:
	void FindPath(BinaryTreeNode* pRoot, int expectedSum, std::vector<int>& path, int currentSum);
	void FindPath(BinaryTreeNode* pRoot, int expectedSum) {
		if (pRoot == nullptr)
			return;

		std::vector<int> path;
		int currentSum = 0;
		FindPath(pRoot, expectedSum, path, currentSum);
	}

	void FindPath(BinaryTreeNode* pRoot, int expectedSum, std::vector<int>& path, int currentSum) {
		currentSum += pRoot->m_mValue;
		path.push_back(pRoot->m_mValue);

		bool isLeaf = pRoot->m_pLeft == nullptr && pRoot->m_pRight == nullptr;
		if (currentSum == expectedSum && isLeaf) {
			std::cout << "A path is found" ;
			std::vector<int>::iterator iter = path.begin();
			for (; iter != path.end(); ++iter)
				std::cout << *iter << "\t";

			std::cout << std::endl;
		}

		if (pRoot->m_pLeft != nullptr)
			FindPath(pRoot->m_pLeft, expectedSum, path, currentSum);
		if (pRoot->m_pRight != nullptr)
			FindPath(pRoot->m_pRight, expectedSum, path, currentSum);

		path.pop_back();
	}
};

猜你喜欢

转载自blog.csdn.net/love__live1/article/details/82966212