LeetCode 113 [Path Sum II]

题目链接

题目描述

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

解题思路

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

C++代码

/**
 * 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();
    }
};