二叉树根节点到叶子节点和为指定值的路径（NC8/考察次数Top54/难度中等）

Posted 2021-07-23 码农指南

描述：
给定一个二叉树和一个值sum，请找出所有的根节点到叶子节点的节点值之和等于sum的路径，
例如：
给出如下的二叉树，sum=22，

返回
[
[5,4,11,2],
[5,8,9]
]

示例1
输入：
{1,2},1
返回值：
[]
(题目来自牛客网)

用C++实现如下

class Solution {
public:
    /**
     *
     * @param root TreeNode类
     * @param sum int整型
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > pathSum(TreeNode* root, int sum) {
        // write code here
        //调用dfs函数来实现
        dfs(root,sum,0);                       //利用递归值来实现
        return res;
    }
     
    void dfs(TreeNode* root, int sum, int cur)
    {
        if(root == NULL)                       //节点为空，直接返回
            return;
        tmp.push_back(root->val);              //把经过的节点值临时存到tmp中
        cur+=root->val;                        //cur用于记录临时的累加和
        if(root->left == NULL && root->right == NULL)
        {
            if(sum == cur)                     //到达叶子节点后进行判断处理
                res.push_back(tmp);
        }else{
            dfs(root->left,sum,cur);           //递归左子树
            //左边子树递归完后，递归右子树
            dfs(root->right,sum,cur);          //递归右子树
        }
        //碰到根节点后，cur减去最后一个，同时tmp弹出删除最后一个
        cur-=tmp[tmp.size()-1];                
        tmp.pop_back();                        
    }
     
private:
    //定义两个成员变量来存储结果res，以及路径过程中的节点值组成的tmp;
    vector<vector<int>> res;                   //存放结果；
    vector<int>tmp;                            //存放临时路径节点值；
};

纯手撕代码，如果觉得内容不错麻烦点个赞，后面陆续配上Top100算法题通俗易懂的讲解视频，可以花两个月时间完全掌握，进大厂不是梦，转行狗亲测！