二叉树的和为某一值的路径(Java)
题目:输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
代码:
import java.util.ArrayList;
/**
class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public static void main(String args[]) {
}
//单个满足条件的路径
ArrayList<Integer> list = new ArrayList<>();
//找到所有满足条件的路径
ArrayList<ArrayList<Integer>> path = new ArrayList<>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
if(root != null) {
list.add(root.val);
//当单个路径加入一个元素,则 target 的值就要减去相应的值。
target -= root.val;
boolean isLeaf = root.left == null && root.right == null;
//从根到叶子的路径上值加和等于target时,才能将给路径放到结果集中。
if (target == 0 && isLeaf) {
path.add(new ArrayList<>(list));
} else {
//不是叶子结点,遍历它的左右结点。
FindPath(root.left, target);
FindPath(root.right, target);
}
//回退的时候,需要删除单个路径中父结点。
list.remove(list.size() - 1);
}
return path;
}
}