1. 程式人生 > 實用技巧 >劍指offer_24:二叉樹中和為某一值的路徑

劍指offer_24:二叉樹中和為某一值的路徑

輸入一棵二叉樹和一個整數,打印出二叉樹中節點值的和為輸入整數的所有路徑。從樹的根節點開始往下一直到葉節點所經過的節點形成一條路徑。

示例:
給定如下二叉樹,以及目標和 sum = 22,

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

提示:
節點總數 <= 10000

1、DFS

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    List<List<Integer>> res=new ArrayList<>();
    LinkedList<Integer> list=new LinkedList<>();
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        dfs(root,sum);
        return res;
    }
    public void dfs(TreeNode node,int sum){
        if(node==null) return;
        list.add(node.val);
        sum-=node.val;
        if(sum==0&&node.left==null&&node.right==null){
            res.add(new LinkedList(list));
        }
        dfs(node.left,sum);
        dfs(node.right,sum);
        list.removeLast();
    }
}