題目描述
輸入一顆二叉樹和一個整數(shù),打印出二叉樹中結(jié)點(diǎn)值的和為輸入整數(shù)的所有路徑。路徑定義為從樹的根結(jié)點(diǎn)開始往下一直到葉結(jié)點(diǎn)所經(jīng)過的結(jié)點(diǎn)形成一條路徑。
本題注意點(diǎn):
- 遞歸結(jié)束后返回父節(jié)點(diǎn)
- ArrayList的拷貝
import java.util.ArrayList;
/**
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>> lists = new ArrayList<ArrayList<Integer>>();
if(root==null){
return lists;
}
ArrayList<Integer> list = new ArrayList<>();
FindPath(root,target,list,lists);
return lists;
}
int curr = 0;
public void FindPath(TreeNode root,int target, ArrayList<Integer> list,ArrayList<ArrayList<Integer>> lists) {
if(root!=null){
curr+=root.val;
list.add(root.val);
}
//到達(dá)葉子節(jié)點(diǎn)
if(root.left==null&&root.right==null){
if(curr==target){
//這里必須new一個list來保存結(jié)果
ArrayList<Integer> list2 = new ArrayList<>();
list2 = (ArrayList<Integer>)list.clone();
lists.add(list2);
}
}
if(root.left != null)
FindPath(root.left,target,list,lists);
if(root.right != null)
FindPath(root.right,target,list,lists);
//遞歸會返回到父節(jié)點(diǎn),在此之前刪掉本節(jié)點(diǎn)
curr-=root.val;
list.remove(list.size()-1);
}
}