题目:
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
思路:
- 由于路径是从根结点出发到叶结点, 也就是说路径总是以根结点为起始点,因此我们首先需要遍历根结点。在树的前序、中序、后序三种遍历方式中,只有前序遍历是首先访问根结点的。
- 当用前序遍历的方式访问到某一结点时, 我们把该结点添加到路径上,并累加该结点的值。
- 如果该结点是叶结点并且路径中结点值的和刚好等于输入的整数, 则当前的路径符合要求,我们把它打印出来。
- 如果当前结点不是叶结点,则继续访问它的子结点。
- 当前结点访问结束后,递归函数将自动回到它的父结点。因此我们在函数退出之前要在路径上删除当前结点并减去当前结点的值,以确保返回父结点时路径刚好是从根结点到父结点的路径。
- 不难看出保存路径的数据结构实际上是一个栈, 因为路径要与递归调用状态一致, 而递归调用的本质就是一个压栈和出栈的过程。
代码:
package 二叉树中和某一值的路径25;
import java.util.ArrayList;
/**
* 输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。
* 路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
* @author Administrator
* 测试用例为完善
* 注意:提交 牛客网 时,方法中的 static 需要删除
*/
class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public class Demo {
// 最终返回的动态数组
private static ArrayList<ArrayList<Integer>> listAll = new ArrayList<ArrayList<Integer>>();
// 用于保存经过的路径上的结点
private static ArrayList<Integer> list = new ArrayList<Integer>();
public static ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
if (root == null) {
return listAll;
}
// 记录当前结点
list.add(root.val);
// 每记录一个结点,计算一次和目标值的差
target -= root.val;
// 当路径的总和与目标值相等并且为叶节点时,将已经记录的结点添加到动态数组中
if (target == 0 && root.left == null && root.right == null) {
listAll.add(new ArrayList<Integer>(list));
} else {
// 如果不是叶节点,则遍历它的子结点
FindPath(root.left, target);
FindPath(root.right, target);
}
// 在返回父结点之前,在路径上删除当前结点
list.remove(list.size() - 1);
return listAll;
}
public static void main(String[] args) {
// 10
// / \
// 5 12
// / \
// 4 7
TreeNode root = new TreeNode(10);
root.left = new TreeNode(5);
root.left.left = new TreeNode(4);
root.left.right = new TreeNode(7);
root.right = new TreeNode(12);
// 有两条路径上的结点和为22
System.out.println("findPath(root, 22);");
FindPath(root, 22);
for (int i = 0; i < listAll.size(); i++) {
System.out.println(listAll);
}
// 没有路径上的结点和为15
System.out.println("findPath(root, 15);");
FindPath(root, 15);
for (int i = 0; i < listAll.size(); i++) {
System.out.println(listAll);
}
// // 有一条路径上的结点和为19
// System.out.println("findPath(root, 19);");
// FindPath(root, 19);
// // 5
// // /
// // 4
// // /
// // 3
// // /
// // 2
// // /
// // 1
// TreeNode root2 = new TreeNode(5);
// root2.left = new TreeNode(4);
// root2.left.left = new TreeNode(3);
// root2.left.left.left = new TreeNode(2);
// root2.left.left.left.left = new TreeNode(1);
// // 有一条路径上面的结点和为15
// System.out.println("findPath(root2, 15);");
// FindPath(root2, 15);
// // 没有路径上面的结点和为16
// System.out.println("findPath(root2, 16);");
// FindPath(root2, 16);
// // 1
// // \
// // 2
// // \
// // 3
// // \
// // 4
// // \
// // 5
// TreeNode root3 = new TreeNode(1);
// root3.right = new TreeNode(2);
// root3.right.right = new TreeNode(3);
// root3.right.right.right = new TreeNode(4);
// root3.right.right.right.right = new TreeNode(5);
// // 有一条路径上面的结点和为15
// System.out.println("findPath(root3, 15);");
// FindPath(root3, 15);
// // 没有路径上面的结点和为16
// System.out.println("findPath(root3, 16);");
// FindPath(root3, 16);
// // 树中只有1个结点
// TreeNode root4 = new TreeNode(1);
// // 有一条路径上面的结点和为1
// System.out.println("findPath(root4, 1);");
// FindPath(root4, 1);
// // 没有路径上面的结点和为2
// System.out.println("findPath(root4, 2);");
// FindPath(root4, 2);
// // 树中没有结点
// System.out.println("findPath(null, 0);");
// FindPath(null, 0);
}
}