#yyds干貨盤點# LeetCode程序員面試金典:求和路徑

2022-12-27 17:30:25 來源:51CTO博客


【資料圖】

題目:

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

示例:給定如下二叉樹,以及目標和?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]

代碼實現:

class Solution {    public int pathSum(TreeNode root, int sum) {        Map prefix = new HashMap();        prefix.put(0L, 1);        return dfs(root, prefix, 0, sum);    }    public int dfs(TreeNode root, Map prefix, long curr, int sum) {        if (root == null) {            return 0;        }        int ret = 0;        curr += root.val;        ret = prefix.getOrDefault(curr - sum, 0);        prefix.put(curr, prefix.getOrDefault(curr, 0) + 1);        ret += dfs(root.left, prefix, curr, sum);        ret += dfs(root.right, prefix, curr, sum);        prefix.put(curr, prefix.getOrDefault(curr, 0) - 1);        return ret;    }}

標簽: 設計一個

上一篇:#yyds干貨盤點# LeetCode程序員面試金典:插入
下一篇:Oracle數據庫中convert()函數,在瀚高數據庫中如何替換使用?