LeetCode 113 [Path Sum II]

题目链接

题目描述

  • 给定一棵二叉树与一个整数值,要求查找出所有从根结点到叶子结点的路径,这些路径上的所有结点的和等于给定的值。

解题思路

  • 题目默认条件

    • 不需要考虑子路径结点的和可能溢出的情况
  • 本题主要关注点在于三个方面

    1. 如何判断某个结点是不是叶子结点
      • 如果该结点没有左子结点与右子结点,则该结点是叶子结点,根结点也可能是叶子结点
    2. 若采用递归的方式,如何减少内存空间的使用
      • 存放临时路径的vector,在整个遍历过程都使用引用来传递该临时路径,每个结点处理之后需要清理对该路径的变更
      • 存放结果的vector,在整个遍历过程中同样使用引用来传递
    3. 如何判断某条路径满足条件
      • 在遍历过程中,用一个变量实时记录临时路径的值,若某个结点是叶子结点,并且该变量的值与给定值相等,则当前的临时路径满足条件,可加入存放结果的vector中

C++代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> pathSum(TreeNode* root, int sum) {
vector<vector<int>> result;
vector<int> subPath;
if (root) {
calculatePathSum(root, sum, subPath, 0, result);
}
return result;
}
void calculatePathSum(TreeNode* node, int sum, vector<int>& subPath, int subSum, vector<vector<int>>& result) {
subPath.push_back(node->val);
subSum += node->val;
if (!node->left && !node->right) {
if (sum == subSum) {
result.push_back(subPath);
}
}
if (node->left) {
calculatePathSum(node->left, sum, subPath, subSum, result);
}
if (node->right) {
calculatePathSum(node->right, sum, subPath, subSum, result);
}
subSum -= node->val;
subPath.pop_back();
}
};