题目描述
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
例如,求路径和为22:
有两条:10->5->7、10->12
由于路径是从根节点开始到叶节点的,要想到是用前序遍历,同时需要记录下当前的路径,当遍历到叶节点时判断即可。
import java.util.ArrayList;
/**
* @author yuan
* @date 2019/2/17
* @description
*/
public class 二叉树中和为某一值的路径 {
static class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
ArrayList<ArrayList<Integer>> result = new ArrayList<>();
dfs(root, target, new ArrayList<>(), 0, result);
return result;
}
/**
* 用前序遍历,因为前序是先访问根节点
* @param root
* @param target
* @param list 记录当前的一条路径
* @param cur 记录当前一条路径的和
* @param result 结果
*/
private void dfs(TreeNode root, int target, ArrayList<Integer> list, int cur, ArrayList<ArrayList<Integer>> result) {
if (root == null) {
return;
}
cur += root.val;
// 如果叶节点
if (root.left == null && root.right == null && cur == target) {
ArrayList<Integer> li = new ArrayList<>(list);
li.add(root.val);
result.add(li);
return;
}
// 把当前节点保存下来,继续遍历节点的左右子树
list.add(root.val);
if (root.left != null) {
dfs(root.left, target, list, cur, result);
}
if (root.right != null) {
dfs(root.right, target, list, cur, result);
}
// 返回上一个父节点前,需要路径上删除当前节点
list.remove(list.size() - 1);
}
}