題目描述

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

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

?

示例1

輸入

{1,2},1

返回值

[]

示例2

輸入

{1,2},3

返回值

[[1,2]]

?

思路：使用遞歸

?從根節點開始，遍歷每一個結點（深度優先遍歷），將該結點的值放入當前ArrayList中

判斷當前結點是否為葉子結點 以及 當前的路徑和是否與sum相等?

步驟2中的兩個條件都滿足時，則將list加入到結果集中。否則回退，刪除list的最后一個元素（該元素可能不是葉節點，也可能當前路徑不等于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.初始當前的列表ArrayList<Integer> pathList = new ArrayList();// 2.深度優先遍歷dfs(allList,pathList,root,sum);//3.返回路徑集合return allList;}//一、對root進行深度優先遍歷//result:所有路徑和為sum的結果集 list：當前路徑結點集 root：當前根節點 sum：當前所需的路徑和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){//為葉子節點if(sum - root.val == 0){//加上該結點后，滿足路徑和sumpathList.add(root.val);//將當前結點加入listallList.add(new ArrayList<>(pathList));//將當前路徑 加入 結果集pathList.remove(pathList.size()-1);//去掉當前結點，繼續下面的遍歷}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);} }

?

?

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。