题解 | #二叉树中和为某一值的路径(三)#

二叉树中和为某一值的路径(三)

http://www.nowcoder.com/practice/965fef32cae14a17a8e86c76ffe3131f

解题思路

层序遍历+递归

借助队列,层序遍历将每个节点添加到队列中,然后依次从队列中取出节点作为子树的根节点,来遍历每颗子树

如果Path与sum相等,那么就可以++res来增加路径数

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @param sum int整型 
     * @return int整型
     */
    int FindPath(TreeNode* root, int sum) {
        // write code here
        if(!root)
            return 0;
        queue<TreeNode*> q;
        //根节点入队
        q.push(root);
        int nCount = 0;//记录每层节点的个数
        res = 0;
        //path = 0;
        while(!q.empty()){
            //记录当前层数的节点
            nCount = q.size();
            while(nCount--){
                TreeNode *node = q.front();
                q.pop();
                //该节点的左右孩子入队
                if(node->left)
                    q.push(node->left);
                if(node->right)
                    q.push(node->right);
                dfs(node,sum,0);
            }
        }
        return res;
    }
    void dfs(TreeNode* root,int sum,int path)
    {
        if(!root)
                return;
        path += root->val;
        if(path == sum)
            ++res;
        //左
        dfs(root->left,sum,path);
        //右
        dfs(root->right,sum,path);
    }
private:
    int res;
    //int path;
};
全部评论

相关推荐

我即大橘:耐泡王
点赞 评论 收藏
分享
沉淀一会:1.同学你面试评价不错,概率很大,请耐心等待; 2.你的排名比较靠前,不要担心,耐心等待; 3.问题不大,正在审批,不要着急签其他公司,等等我们! 4.预计9月中下旬,安心过节; 5.下周会有结果,请耐心等待下; 6.可能国庆节前后,一有结果我马上通知你; 7.预计10月中旬,再坚持一下; 8.正在走流程,就这两天了; 9.同学,结果我也不知道,你如果查到了也告诉我一声; 10.同学你出线不明朗,建议签其他公司保底! 11.同学你找了哪些公司,我也在找工作。
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务