剑指 Offer 34. 二叉树中和为某一值的路径(中等)
题目:
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void traversal(TreeNode* cur, int sum){
if(!cur->left&&!cur->right&&sum==0){ //如果是叶子结点,且此时和减去该节点值为0了,则将该条路径添加到result中
result.push_back(path);
return;
}
if(!cur->left&&!cur->right){ //如果是叶子结点,但和不为0,则返回
return;
}
if(cur->left){
path.push_back(cur->left->val); //若左节点存在,将其添加到path中
traversal(cur->left,sum-cur->left->val); //对左节点进行递归,注意传入的是sum-cur->left->val,所以sum的值在本次递归中没有被改变。包含了回溯的思想
path.pop_back(); //回溯path数组
}
if(cur->right){ //若右节点存在,做相同处理
path.push_back(cur->right->val);
traversal(cur->right,sum-cur->right->val);
path.pop_back();
}
return;
}
vector<vector<int>> pathSum(TreeNode* root, int target) {
result.clear();
path.clear();
if(!root) return result;
path.push_back(root->val); //首先添加根节点的值到path中
traversal(root, target-root->val);
return result;
}
};
以上方法来自代码随想录