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

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

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;
};
全部评论

相关推荐

杨柳哥:这不是普通人,那这个钱的是天才
点赞 评论 收藏
分享
hso_:哈哈哈哈哈哈我没offer一样在同一道题开喷了
投递深圳同为数码等公司10个岗位
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务