描述
输入一颗二叉树的根节点root和一个整数expectNumber,找出二叉树中结点值的和为expectNumber的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
如二叉树root为{10,5,12,4,7},expectNumber为22
则合法路径有[[10,5,7],[10,12]]
数据范围:
树中节点总数在范围 [0, 5000] 内
-1000 <= 节点值 <= 1000
-1000 <= expectNumber <= 1000
示例1
输入:
{10,5,12,4,7},22
返回值:
[[10,5,7],[10,12]]
说明:
返回[[10,12],[10,5,7]]也是对的
示例2
输入:
{10,5,12,4,7},15
返回值:
[]
示例3
输入:
{5,4,8,11,#,13,4,7,2,#,#,5,1},22
返回值:
[[5,4,11,2],[5,8,4,5]]
解题思路:
根据题意,由跟节点至叶节点,路径之和为参数值。则判断条件为:根节点开始,叶节点结束,和为参数值。
找出路径需要遍历整棵树,这里采用先序遍历,即:根节点,左子树,右子树。
递归算法三部曲:
明白递归函数的功能:FindPath(TreeNode root,int expectNumber),从root节点出发,找和为expectNumber的路径
递归终止条件:当root节点为叶子节点并且expectNumber==root.val
下一次递归:如果左子树不空,递归左子树;如果右子树不空,递归右子树;
但是,,这里没有保存路径啊?可以用一个全局变量ArrayList<ArrayList> res = new ArrayList<>();
代码如下:
import java.util.ArrayList;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
//全局变量
ArrayList<ArrayList<Integer>> res = new ArrayList<>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int expectNumber) {
if(root == null) return res;
helper(root, expectNumber,new ArrayList<Integer>());
return res;
}
public void helper(TreeNode root, int k, ArrayList<Integer> temp){
//将root节点放入路径
temp.add(root.val);
//判断条件:目标值&&叶节点
if(root.val == k && root.left == null && root.right == null){
//new一个新的ArrayList放入,否则当temp改变时,res会跟着变
res.add(new ArrayList(temp));
}
//分别递归左右子树
if(root.left != null) helper(root.left, k - root.val, temp);
if(root.right != null) helper(root.right, k - root.val, temp);
//回溯时删除当前节点,该节点不应该在下一条可能的路径中
temp.remove(temp.size() - 1);
}
}
复杂度分析:
时间复杂度:O(N^2),其中 N 是树的节点数。在最坏情况下,树的上半部分为链状,下半部分为完全二叉树,并且从根节点到每一个叶子节点的路径都符合题目要求。此时,路径的数目为 O(N),并且每一条路径的节点个数也为 O(N),因此要将这些路径全部添加进答案中,时间复杂度为 O(N^2)
空间复杂度:O(N),其中 N 是树的节点数。空间复杂度主要取决于返回列表空间的开销,元素个数不会超过树的节点数