【在线编程】二叉树中和为某一路径的值
【问题描述】
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
【解题思路 & Java实现】
思路:深度遍历二叉树,利用path记录每条路径,在左右子树中递归寻找满足条件的路径。
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
ArrayList<ArrayList<Integer>> result = new ArrayList<>();
ArrayList<Integer> path = new ArrayList<>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
if (root == null || target < root.val) {
return result;
}
path.add(root.val);
target -= root.val;
//如果当前结点为叶子结点,并且满足路径和为target
if (target == 0 && root.left == null && root.right == null) {
result.add(new ArrayList<Integer>(path));
}
if (root.left != null) {
FindPath(root.left, target);
}
if (root.right != null) {
FindPath(root.right, target);
}
//path回退一步,寻找另一条路径
path.remove(path.size() - 1);
//Comparator,先确定比较单位是什么,然后升序,o1在前,降序,o2在前
Collections.sort(result, new Comparator<ArrayList<Integer>>() {
@Override
public int compare(ArrayList<Integer> o1, ArrayList<Integer> o2) {
return o2.size() - o1.size();
}
});
return result;
}
}