leetcode437 路和總徑3

給定一個二叉樹的根節(jié)點(diǎn) root ，和一個整數(shù) targetSum ，求該二叉樹里節(jié)點(diǎn)值之和等于 targetSum 的 路徑 的數(shù)目。

路徑 不需要從根節(jié)點(diǎn)開始，也不需要在葉子節(jié)點(diǎn)結(jié)束，但是路徑方向必須是向下的（只能從父節(jié)點(diǎn)到子節(jié)點(diǎn)）。

示例 1：

輸入：root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
輸出：3
解釋：和等于 8 的路徑有 3 條，如圖所示。

文章目錄

• leetcode437 路和總徑3
• 一、遞歸（時間復(fù)雜度高）
• 二、前綴和

一、遞歸（時間復(fù)雜度高）

class Solution {int res=0;//前序遍歷每個二叉樹節(jié)點(diǎn)public int pathSum(TreeNode root, int targetSum) {if(root==null)return 0;curNodeSum(root,targetSum);pathSum(root.left,targetSum);pathSum(root.right,targetSum);return res;}//針對每個節(jié)點(diǎn)為首節(jié)點(diǎn)求出一連串和為targetSum的子樹有多少對public void curNodeSum(TreeNode root,int targetSum){if(root==null){return;}targetSum=targetSum-root.val;if(targetSum==0){res++;}curNodeSum(root.left,targetSum);curNodeSum(root.right,targetSum);} }

二、前綴和

class Solution{//前綴和思想： Key:和為key value:和為key的路徑有多少條HashMap<Integer,Integer> map=new HashMap<>();public int pathSum(TreeNode root,int targetSum){if(root==null)return 0;map.put(0,1);return getSum(root,targetSum,0);}//這邊必須是前序遍歷public int getSum(TreeNode root,int targetSum,int sum){if(root==null)return 0;int res=0;//求出到當(dāng)前節(jié)點(diǎn)的路徑和int curPathSum=sum+root.val;//看之前是否存在以curPathSum-targetSum為key的前綴和路徑res+=map.getOrDefault(curPathSum-targetSum,0);//保存當(dāng)前前綴和數(shù)量map.put(curPathSum,map.getOrDefault(curPathSum,0)+1);res+=getSum(root.left,targetSum,sum+root.val);res+=getSum(root.right,targetSum,sum+root.val);//進(jìn)行回溯map.put(curPathSum,map.get(curPathSum)-1);return res;} }

總結(jié)

以上是生活随笔為你收集整理的1、leetcode437 路和总径3的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。