二叉树根节点到叶子节点路径的问题考察形式很多,这道是和为指定值的路径,我们使用递归,在遍历的过程中,不仅路径上的累加和要等于目标值,必须是到达叶子节点的路径和,所以判断时,我们要满足三个条件,前两个条件是当前节点的左右子树都为空(说明是叶子节点),最后一个条件是当前节点的值要等于传递的sum值。因为我们传递的sum值是目标值减去前面所有节点和,若当前节点满足三个条件,说明此条路径满足要求,我们将这条路径的节点值都存在vec中,最后将这条路径放入res中。继续递归直到所有路径遍历完毕。
求二叉树根节点到叶子节点的所有路径和,可以参考这篇博客。
注:vec用来保存路径上的节点值,res就是最后所有满足条件的路径合集。
代码如下:结合代码理解
class Solution {
public:
/**
*
* @param root TreeNode类
* @param sum int整型
* @return int整型vector<vector<>>
*/
vector<vector<int> > pathSum(TreeNode* root, int sum) {
// write code here
vector<vector<int> >res;
vector<int>vec;
if(root==nullptr) return res;
dfs(root,sum,vec,res);
return res;
}
void dfs(TreeNode* root,int sum,vector<int>vec,vector<vector<int>>&res)
{
if(root==nullptr)
return;
vec.push_back(root->val);
if(root->left==nullptr&&root->right==nullptr&&root->val==sum)
res.push_back(vec);
dfs(root->left,sum-root->val,vec,res);
dfs(root->right,sum-root->val,vec,res);
}
};