題目:輸入一顆二叉樹的跟節點和一個整數,打印出二叉樹中結點值的和爲輸入整數的所有路徑。路徑定義爲從樹的根結點開始往下一直到葉結點所經過的結點形成一條路徑。(注意: 在返回值的list中,數組長度大的數組靠前)
思路:
遞歸先序遍歷樹, 把結點加入路徑。
若該結點是葉子結點則比較當前路徑和是否等於期待和。
彈出結點,每一輪遞歸返回到父結點時,當前路徑也應該回退一個結點
Java代碼
import java.util.ArrayList;
/*
//TreeNode也可以寫成內部類
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
ArrayList<ArrayList<Integer>> arrlist = new ArrayList<ArrayList<Integer>>();
if(root == null) return arrlist;
ArrayList<Integer> a1 = new ArrayList<Integer>();
int sum = 0;
sums(root, target, arrlist,a1, sum);
return arrlist;
}
public void sums(TreeNode root, int target, ArrayList<ArrayList<Integer>> arrlist, ArrayList<Integer> a1, int sum){
if(root == null) return;
sum = sum + root.val;
if(root.left == null && root.right == null){
if(sum == target){
a1.add(root.val);
arrlist.add(new ArrayList<Integer>(a1));
//
a1.remove(a1.size()-1);
}
return;
}
a1.add(root.val);
sums(root.left, target, arrlist,a1, sum);
sums(root.right, target, arrlist,a1, sum);
a1.remove(a1.size()-1);
}
}
更簡潔的寫法
// super nice
public class Solution{
private ArrayList<ArrayList<Integer>> listAll = new ArrayList<ArrayList<Integer>();
private ArrayList<Integer> list = new ArrayList<Integer>();
public ArrayList<ArrayList<Integer>> FindPaths(TreeNode root,int target) {
if(root == null) return listAll;
list.add(root.val);
target -= root.val;
if(target == 0 && root.left == null && root.right == null)
listAll.add(new ArrayList<Integer>(list));
FindPath(root.left, target);
FindPath(root.right, target);
list.remove(list.size()-1);
return listAll;
}
//構建一個 TreeNode 內部類
class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}