1. 程式人生 > 實用技巧 >HDU2883 kebab (壓縮區間+網路流)

HDU2883 kebab (壓縮區間+網路流)

給定一棵二叉樹,其中每個節點都含有一個整數數值(該值或正或負)。設計一個演算法,列印節點數值總和等於某個給定值的所有路徑的數量。注意,路徑不一定非得從二叉樹的根節點或葉節點開始或結束,但是其方向必須向下(只能從父節點指向子節點方向)。

示例:
給定如下二叉樹,以及目標和sum = 22

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

返回:

3
解釋:和為 22的路徑有:[5,4,11,2], [5,8,4,5], [4,11,7]

提示:

  • 節點總數 <= 10000
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void dfs(TreeNode* root,int sum,int &curSum,int &pathNum){
        
if(!root)return; curSum+=root->val; if(curSum==sum)pathNum++; if(root->left)dfs(root->left,sum,curSum,pathNum); if(root->right)dfs(root->right,sum,curSum,pathNum); curSum-=root->val; } int pathSum(TreeNode* root, int sum) { if
(!root)return 0; int curSum=0,pathNum=0; dfs(root,sum,curSum,pathNum); pathNum+=pathSum(root->left,sum); pathNum+=pathSum(root->right,sum); return pathNum; } };