劍指 Offer - 24:二叉樹中和為某一值的路徑
阿新 • • 發佈:2018-12-18
題目描述
輸入一顆二叉樹的跟節點和一個整數,打印出二叉樹中結點值的和為輸入整數的所有路徑。路徑定義為從樹的根結點開始往下一直到葉結點所經過的結點形成一條路徑。(注意: 在返回值的list中,陣列長度大的陣列靠前)
題目連結:https://www.nowcoder.com/practice/b736e784e3e34731af99065031301bca
public class Solution {
ArrayList<ArrayList<Integer>> result = new ArrayList<>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
if (root == null) {
return result;
}
int root_val = root.val;
ArrayList<Integer> list = new ArrayList<>();
list.add(root_val);
rec(root, list, root_val, target);
Collections.sort(result, (o1, o2) -> o2.size() - o1.size());
return result;
}
private void rec(TreeNode node, ArrayList<Integer> list, int sum, int target) {
if (node.left == null && node.right == null) {
if (sum == target) {
result.add(new ArrayList<>(list));
}
return;
}
// 遞迴左子樹
if (node.left != null) {
Integer value = node.left.val;
list.add(value);
rec(node.left, list, sum+value, target);
list.remove(value);
}
// 遞迴右子樹
if (node.right != null) {
Integer value = node.right.val;
list.add(value);
rec(node.right, list, sum+value, target);
list.remove(value);
}
}
}