113. 路径总和 II(递归、双层集合List、双向链表LinkedList实现栈)
https://leetcode-cn.com/problems/path-sum-ii/
题目**
给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
思想**
路径总和 I的思想,再进一步改进,具体见解析和注释
求路径消耗总和为指定数量的路径,不同于I的判定,II要求直接存储返回各个路径,所以在I的基础上要声明一个result存储各个路径,其中每个元素为一条路径,而又得开辟一个双向链表栈stack来存储(便于增删)当前路径
代码**
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
/*
路径总和 I的思想,再进一步改进
求路径消耗总和为指定数量的路径,不同于I的判定,II要求直接存储返回各个路径,所以在I的基础上要声明一个result存储各个路径,其中每个元素为一条路径,而该路径又得开辟一个链表数组stack来存储(便于增删)
*/
class Solution {
public void pathSum(TreeNode root, int sum, List<List<Integer>> result, LinkedList<Integer> stack){
if(root == null){
return;
}
stack.addLast(root.val);//将当前结点加入当前路径
if(root.left == null && root.right == null && root.val == sum){
result.add(new ArrayList<Integer>(stack));
}
pathSum(root.left, sum - root.val, result, stack);//左子树,若root为根结点,返回null
pathSum(root.right, sum - root.val, result, stack);//右子树,若root为根结点,返回null
stack.removeLast();//将当前结点退出当前路径,寻找下一条符合条件路径(回溯思想)
}
public List<List<Integer>> pathSum(TreeNode root, int sum) {
List<List<Integer>> result = new ArrayList<>();//result存储各个路径
LinkedList<Integer> stack = new LinkedList<>();//链表数组stack来存储(便于增删)当前路径
pathSum(root, sum, result, stack);
return result;
}
}