HDU2883 kebab (壓縮區間+網路流)
阿新 • • 發佈:2020-11-24
給定一棵二叉樹,其中每個節點都含有一個整數數值(該值或正或負)。設計一個演算法,列印節點數值總和等於某個給定值的所有路徑的數量。注意,路徑不一定非得從二叉樹的根節點或葉節點開始或結束,但是其方向必須向下(只能從父節點指向子節點方向)。
示例:
給定如下二叉樹,以及目標和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; } };