剑指 offer:二叉树中和为某一值的路径

本文介绍了一种使用递归算法解决二叉树路径求和问题的方法。通过递归遍历二叉树并记录路径节点,找出所有节点值之和等于指定整数的路径。

题目描述
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)

分析
树经常是和递归联系在一起的,所以,这题也要用到递归算法。
问题就在于如何保存元素。我最开始的想法是用栈来保存路径节点,但在最后将栈存到 vector 中的时候出现了问题:要将栈中所有元素清空,而我不希望栈被清空,所以,我将 stack 直接换成了vector,这样就省事了。
具体过程比较难说,我尽量说吧。
每次递归进行到当前节点,将当前节点存进数组(下面代码中的 B),再递归遍历其左右子节点。在遍历完左子节点之后,需要将左子节点弹出来,因为要继续遍历右子节点(每次递归时,只需要考虑下一轮的情况即可,就像每个人只需管好自己的事,整个工作就可以做得很好)。遍历完右子节点后,也要将其弹出,因为当前节点的祖先还要继续工作。总的来说,就是每次递归完当前节点之后,就要将当前节点的影响消除,不要影响到其他节点(当前节点的兄弟节点或其祖先节点)的遍历。
整体思路就是这样了。
代码如下:

void ifSum(vector<vector<int> > &A, vector<int> &B, int sum, int expectNumber, TreeNode* p) {
	if (!p) return;
	if (sum + p->val == expectNumber && !p->left && !p->right) {
		B.push_back(p->val);
		//push into vector
		A.push_back(B);
		//return 1;
	}
	if (sum + p->val < expectNumber) {
		B.push_back(p->val);
		ifSum(A, B, sum + p->val, expectNumber, p->left);
		if (p->left)
			B.pop_back();
		ifSum(A, B, sum + p->val, expectNumber, p->right);
		if (p->right)
			B.pop_back();
	}
}

vector<vector<int> > FindPath(TreeNode* root, int expectNumber) {
	vector<vector<int> > A;
	vector<int> B;
	ifSum(A, B, 0, expectNumber, root);
	return A;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值