题目描述
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的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;
}
本文介绍了一种使用递归算法解决二叉树路径求和问题的方法。通过递归遍历二叉树并记录路径节点,找出所有节点值之和等于指定整数的路径。

9445

被折叠的 条评论
为什么被折叠?



