题解 | #牛牛的路径总和#
牛牛的路径总和
https://www.nowcoder.com/practice/45a878da071d43458b8b4cb2db8275b6
/** * struct TreeNode { * int val; * struct TreeNode *left; * struct TreeNode *right; * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * }; */ #include <algorithm> #include <cstddef> class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param root TreeNode类 * @param targetSum int整型 * @return int整型vector<vector<>> */ void backtravel(vector<vector<int>>& res, vector<int>& path, TreeNode* root, int& targetSum) { if(root == nullptr){ if (targetSum == 0) res.push_back(path); return; } path.push_back(root->val); targetSum -= root->val; if (root->left == nullptr) { backtravel(res, path, root->right, targetSum); if (root->right != nullptr) { path.pop_back(); targetSum += root->right->val; } } else if (root->right== nullptr) { backtravel(res, path, root->left, targetSum); if (root->left != nullptr) { path.pop_back(); targetSum += root->left->val; } } // 厚大 else if(root->left->val < root->right->val){ backtravel(res, path, root->left, targetSum); path.pop_back(); targetSum += root->left->val; backtravel(res, path, root->right, targetSum); path.pop_back(); targetSum += root->right->val; } else{ backtravel(res, path, root->right, targetSum); path.pop_back(); targetSum += root->right->val; backtravel(res, path, root->left, targetSum); path.pop_back(); targetSum += root->left->val; } return; } vector<vector<int>> findPaths(TreeNode* root, int targetSum) { // write code here vector<vector<int>> res; vector<int> path; backtravel(res, path, root, targetSum); return res; } };