PS:《剑指offer》是很多同学找工作都会参考的一本面试指南,同时也是一本算法指南(为什么它这么受欢迎,主要应该是其提供了一个循序渐进的优化解法,这点我觉得十分友好)。现在很多互联网的算法面试题基本上可以在这里找到影子,为了以后方便参考与回顾,现将书中例题用Java实现(第二版),欢迎各位同学一起交流进步。
GitHub: https://github.com/Uplpw/SwordOffer。
剑指offer完整题目链接: https://blog.csdn.net/qq_41866626/article/details/120415258
1 题目描述
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例:
给定如下二叉树,以及目标和 target = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
leetcode链接: 二叉树中和为某一值的路径(以下代码已测试,提交通过)
2 测试用例
一般是考虑功能用例,特殊(边缘)用例或者是反例,无效测试用例这三种情况。甚至可以从测试用例寻找一些规律解决问题,同时也可以让我们的程序更加完整鲁棒。
(1)功能用例:完全与非完全二叉树。
(2)边缘用例:只有一个节点,只有左节点,只有右节点。
(3)无效用例:树为空。
3 思路
分析:
注意题中的信息:从根节点到叶子节点之和,而不是到某个节点之和,因此必须要遍历二叉树,可以采用深度优先遍历(先序遍历),记录每条路径之和,与期望值相等则加入list。
4 代码
树的结构:
public class TreeNode {
public int val;
public TreeNode left;
public TreeNode right;
public TreeNode(int val) {
this.val = val;
this.left = null;
this.right = null;
}
}
算法实现:
import java.util.ArrayList;
import java.util.List;
public class FindPath {
public static List<List<Integer>> pathSum(TreeNode root, int target) {
List<List<Integer>> list = new ArrayList<>();
if (root == null) {
return list;
}
List<Integer> current = new ArrayList<>();
pathSumCore(root, target, list, current, 0);
return list;
}
public static void pathSumCore(TreeNode root, int target, List<List<Integer>> list, List<Integer> current, int value) {
current.add(root.val);
value = value + root.val;
// 先递归左右子树,直至访问到叶子节点再计算路径之和
if (root.left != null)
pathSumCore(root.left, target, list, current, value);
if (root.right != null)
pathSumCore(root.right, target, list, current, value);
// 符合要求则加入
if (value == target && root.left == null && root.right == null) {
// 由于current是引用类型,在下面操作后,即使加入list也会发生变化,所以要创建新的对象
List<Integer> addList = new ArrayList<>();
for (int i = 0; i < current.size(); i++) {
addList.add(current.get(i));
}
list.add(addList);
}
// 移除最后一个节点,防止对其他节点有影响。
current.remove(current.size() - 1);
}
}
参考
在解决本书例题时,参考了一些大佬的题解,比如leetcode上的官方、K神,以及其他的博客,在之后的每个例题详解后都会给出参考的思路或者代码链接,同学们都可以点进去看看!
本例题参考:
本文如有什么不足或不对的地方,欢迎大家批评指正,最后希望能和大家一起交流进步、拿到心仪的 offer !!!