算法题:剑指 Offer 34. 二叉树中和为某一值的路径 时空 1ms击败99.98%用户(题目+思路+代码+注释)
题目
剑指 Offer 34. 二叉树中和为某一值的路径
给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。
示例 1:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:[[5,4,11,2],[5,8,4,5]]
示例 2:
输入:root = [1,2,3], targetSum = 5
输出:[]
示例 3:
输入:root = [1,2], targetSum = 0
输出:[]
提示:
树中节点总数在范围 [0, 5000] 内
-1000 <= Node.val <= 1000
-1000 <= targetSum <= 1000
注意:本题与主站 113 题相同:https://leetcode-cn.com/problems/path-sum-ii/
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
题目要求找出sum为target的所有路径,那我们想,找一条路径是不是从根节点开始往下遍历,直到叶子结点看是不是等于target(由于题目中有节点和target均可以为负数因此不做剪枝提前返回),因此我们可以在遍历一次这个二叉树就找到所有叶子节点是不是整条路径的sum等于target,但是问题来了,当我们遍历到了叶子节点的时候我没办法知道整条路径了,用数组存到形参里吗?那每次递归之前要拷贝数组,浪费空间,这种事最适合指针了。因此我建立了我的类MyTreeNode 用来完成儿子找爸爸叶子节点向上找根节点找到整个路径的操作。即可只遍历一次就完成!代码分为三块,一个是对外暴露的方法,一个是我的递归遍历二叉树的方法,一个是到了子节点逆推拿路径的方法。简单清晰美观,代码如诗!
代码
public class Solution {
class MyTreeNode {
TreeNode treeNode;
MyTreeNode parent;
public MyTreeNode(TreeNode treeNode, MyTreeNode parent) {
this.treeNode = treeNode;
this.parent = parent;
}
}
public List<List<Integer>> pathSum(TreeNode root, int target) {
//递归遍历搜索树,逆向拿序列即可
List<List<Integer>> list = new ArrayList<>();
if (root != null) {
find(new MyTreeNode(root, null), 0, target, list);
}
return list;
}
/**
* 对根节点向下找
*
* @param node
* @param sum
* @param target
* @param list
*/
public void find(MyTreeNode node, int sum, int target, List<List<Integer>> list) {
//拿到题目中给定的TreeNode
TreeNode treeNode = node.treeNode;
sum += treeNode.val;
//尝试往左节点走
if (treeNode.left != null) {
find(new MyTreeNode(treeNode.left, node), sum, target, list);
}
//尝试往右节点走
if (treeNode.right != null) {
find(new MyTreeNode(treeNode.right, node), sum, target, list);
}
//叶子节点,且和为目标值了,则把这条路径加入结果集
if (treeNode.left == null && treeNode.right == null && sum == target) {
list.add(getPathList(node));
}
}
/**
* 传入叶子节点逆推拿到整条路径
*
* @param thief
* @return
*/
public List<Integer> getPathList(MyTreeNode thief) {
List<Integer> list = new ArrayList<>();
while (thief != null) {
list.add(thief.treeNode.val);
//递推向上走
thief = thief.parent;
}
//翻转
Collections.reverse(list);
return list;
}
}
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode() {
}
TreeNode(int val) {
this.val = val;
}
TreeNode(int val, TreeNode left, TreeNode right) {
this.val = val;
this.left = left;
this.right = right;
}
}
本文来自博客园,作者:HumorChen99,转载请注明原文链接:https://www.cnblogs.com/HumorChen/p/18039491