给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。
输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:[[5,4,11,2],[5,8,4,5]]
解:
和上一个题一样,只不过需要都遍历一次,所以回溯函数的返回值就设为void了。回溯回溯,永远滴神
class Solution {
public:
vector<int> path;
vector<vector<int>> result;
vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
if (!root) return result;
path.push_back(root -> val);
backstracking(root, targetSum - root -> val);
return result;
}
void backstracking(TreeNode* &root, int targetSum) {
if (!root -> left && !root -> right && !targetSum) {
result.push_back(path);
return ;
}
if (!root -> left && !root -> right) {
return ;
}
if (root -> left) {
int nCount = root -> left -> val;
path.push_back(nCount);
backstracking(root -> left, targetSum - nCount);
path.pop_back();
}
if (root -> right) {
int nCount = root -> right -> val;
path.push_back(nCount);
backstracking(root -> right, targetSum - nCount);
path.pop_back();
}
}
};