題目描述

給定一個二叉樹和一個值sum，請找出所有的根節(jié)點(diǎn)到葉子節(jié)點(diǎn)的節(jié)點(diǎn)值之和等于sum?的路徑，
例如：
給出如下的二叉樹，sum=22，

返回
[
[5,4,11,2],
[5,8,9]
]

?

示例1

輸入

{1,2},1

返回值

[]

示例2

輸入

{1,2},3

返回值

[[1,2]]

?

思路：使用遞歸

?從根節(jié)點(diǎn)開始，遍歷每一個結(jié)點(diǎn)（深度優(yōu)先遍歷），將該結(jié)點(diǎn)的值放入當(dāng)前ArrayList中

判斷當(dāng)前結(jié)點(diǎn)是否為葉子結(jié)點(diǎn) 以及 當(dāng)前的路徑和是否與sum相等?

步驟2中的兩個條件都滿足時，則將list加入到結(jié)果集中。否則回退，刪除list的最后一個元素（該元素可能不是葉節(jié)點(diǎn)，也可能當(dāng)前路徑不等于sum）

import java.util.ArrayList;public class TreeNodePutSum {private ArrayList<ArrayList<Integer>> allList = new ArrayList();public static void main(String[] args) {}/**** @param root TreeNode類* @param sum int整型* @return int整型ArrayList<ArrayList<>>*/public ArrayList<ArrayList<Integer>> pathSum (TreeNode root, int sum) {// 樹空if(root==null){return allList;}//1.初始當(dāng)前的列表ArrayList<Integer> pathList = new ArrayList();// 2.深度優(yōu)先遍歷dfs(allList,pathList,root,sum);//3.返回路徑集合return allList;}//一、對root進(jìn)行深度優(yōu)先遍歷//result:所有路徑和為sum的結(jié)果集 list：當(dāng)前路徑結(jié)點(diǎn)集 root：當(dāng)前根節(jié)點(diǎn) sum：當(dāng)前所需的路徑和public void dfs(ArrayList<ArrayList<Integer>> allList,ArrayList<Integer> pathList, TreeNode root, int sum){if(null == root){return;}if(null == root.left && null == root.right){//為葉子節(jié)點(diǎn)if(sum - root.val == 0){//加上該結(jié)點(diǎn)后，滿足路徑和sumpathList.add(root.val);//將當(dāng)前結(jié)點(diǎn)加入listallList.add(new ArrayList<>(pathList));//將當(dāng)前路徑 加入 結(jié)果集pathList.remove(pathList.size()-1);//去掉當(dāng)前結(jié)點(diǎn)，繼續(xù)下面的遍歷}return;}pathList.add(root.val);dfs(allList,pathList, root.left, sum-root.val);dfs(allList,pathList, root.right, sum-root.val);pathList.remove(pathList.size()-1);} }

?

?

總結(jié)

以上是生活随笔為你收集整理的常考数据结构与算法----给定一个二叉树和一个值 sum，请找出所有的根节点到叶子节点的节点值之和等于sum 的路径，的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

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