企业🤖AI智能体构建引擎,智能编排和调试,一键部署,支持私有化部署方案 广告
## 一.题目描述 Given a binary tree and a sum, find all root-to-leaf paths where each path’s sum equals the given sum. For example:  Given the below binary tree and `sum = 22`, ~~~ 5 / \ 4 8 / / \ 11 13 4 / \ / \ 7 2 5 1 ~~~ return ~~~ [ [5,4,11,2], [5,8,4,5] ] ~~~ ## 二.题目分析 这道题与Path Sum类似,不同的是这道题要求找到所有和等于给定值的路径,必须遍历完所有的路径。在找到满足的路径之后,不能直接返回,而是将其添加到一个`vector<vector<int>>`中。在查找的过程中,每经过一个结点,先使用一个`vector<int>`将该路径中的所有结点记录下来。需要注意的是,在进入每一个结点的时候,先将该结点的值`push`到`vector`中,在退出时将该结点的值`pop`出来,这样就可以避免有时会忘记`pop`结点的值的情况。 该方法的时间复杂度为O(n),空间复杂度为O(logn)。 ## 三.示例代码 ~~~ 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> TmpResult; pathSum(root, sum, Result, TmpResult); return Result; } private: void pathSum(TreeNode* root, int sum, vector<vector<int> > &Result, vector<int> &TmpResult) { if (!root) return; if (!root->left && !root->right && root->val == sum) { TmpResult.push_back(sum); Result.push_back(TmpResult); // pop the leaf node TmpResult.pop_back(); return; } int SumChild = sum - root->val; TmpResult.push_back(root->val); pathSum(root->left, SumChild, Result, TmpResult); pathSum(root->right, SumChild, Result, TmpResult); TmpResult.pop_back(); } }; ~~~ ## 四.小结 与Path Sum一样,用DFS解决。