[刷题] 437 Paths Sum III

要求

• 给出一棵二叉树及一个数字sum，判断这棵二叉树上存在多少条路径，其路径上的所有节点和为sum
• 路径不一定始于根节点，终止于叶子节点
• 路径要一直向下

思路

• 分情况讨论：根节点在路径上（8） / 根节点不在路径上（9-10）
• 递归嵌套递归

实现

class Solution {
public:
    int pathSum(TreeNode* root, int sum) {
        
        if( root == NULL )
            return 0;
        
        int res = findPath( root , sum );
        res += pathSum( root->left , sum );
        res += pathSum( root->right , sum );
        
        return res;
    }
    
private:
    // 根节点在路径上 
    int findPath( TreeNode* node, int num ){
        
        if( node == NULL )
            return 0;
        
        int res = 0;
        if( node->val == num )
            res += 1;
        
        res += findPath( node->left , num - node->val );
        res += findPath( node->right , num - node->val );
        
        return res;
        
    }
};