34.二叉樹中和爲某一值的路徑
面試題34. 二叉樹中和爲某一值的路徑
難度中等27
輸入一棵二叉樹和一個整數,打印出二叉樹中節點值的和爲輸入整數的所有路徑。從樹的根節點開始往下一直到葉節點所經過的節點形成一條路徑。
示例:
給定如下二叉樹,以及目標和 sum = 22,
5 / \ 4 8 / / \ 11 13 4 / \ / \ 7 2 5 1
返回:
[ [5,4,11,2], [5,8,4,5] ]
//回溯
// 遞歸解決 終止條件節點爲null 返回
// 當target=0 並且左右子節點都爲null的時候 進行數據添加。
// 否則的話 直接遞歸左右子節點。
List<List> ret = new ArrayList<>();
public List<List<Integer>> pathSum(TreeNode root, int sum) {
pathSum(root,sum,new ArrayList());
return ret;
}
public void pathSum(TreeNode root,int target,List<Integer> path){
if(root == null){
return;
}
path.add(root.val);
target -= root.val;
if(target == 0 && root.left == null && root.right == null){
ret.add(new ArrayList<>(path));
}else{
pathSum(root.left,target,path);
pathSum(root.right,target,path);
}
//回溯,狀態重置,只添加不刪除就是錯的路徑了,因爲整個遞歸過程用的是同一個收集器path
path.remove(path.size()-1);
}