题解 | #二叉树中和为某一值的路径(三)# 递归和层序遍历

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

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

/**
 * 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 res = 0;
    void recursion(TreeNode* node, int sum){
        if(node == nullptr) return;
        if(sum - node->val == 0){
            res++;
        }
        recursion(node->left, sum - node->val);
        recursion(node->right, sum - node->val);
    }
    int FindPath(TreeNode* root, int sum) {
        // write code here
        queue<TreeNode*> que;
        if(root == nullptr) return res;
        que.push(root);
        while(!que.empty()){
            int n = que.size();
            for(int i=0 ;i<n; i++){
                TreeNode *tmp = que.front();
                que.pop();
                if(tmp->left) que.push(tmp->left);
                if(tmp->right) que.push(tmp->right);
                recursion(tmp, sum);
            }
        }
        return res;
    }
};

全部评论

相关推荐

不愿透露姓名的神秘牛友
11-21 19:05
点赞 评论 收藏
分享
one_t:硕还是本?什么岗
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务