タイトル説明
二分木と値の合計が与えられた場合、ルートノードからリーフノードまでのノード値の合計が合計に等しいパスを見つけてください。
たとえば
、次の二分木が与えられた場合、sum = 22、
[
[5,4,11,2]、
[5,8,9]
]を返します
例1
入る
{1,2}、1
戻り値
[]
例2
入る
{1,2}、3
戻り値
[[1,2]]
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* }
*/
public class Solution {
ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> path = new ArrayList();
/**
*
* @param root TreeNode类
* @param sum int整型
* @return int整型ArrayList<ArrayList<>>
*/
public ArrayList<ArrayList<Integer>> pathSum (TreeNode root, int sum) {
preTree(root,sum);
return result;
}
public void preTree(TreeNode root,int sum){
if(root == null){
return;
}
if(root.left == null && root.right == null && sum - root.val == 0){
path.add(root.val);
result.add(new ArrayList<Integer>(path));
path.remove(path.size()-1);
return;
}
path.add(root.val);
preTree(root.left,sum - root.val);
preTree(root.right,sum - root.val);
path.remove(path.size()-1);
}
}