LeetCode113 路径总和II
题目
给你二叉树的根节点 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
方法
深度优先遍历
注意:题目是求从根节点到叶子节点的路径和等于target
- 时间复杂度:O(n^2)?,n为节点个数
- 空间复杂度:O(n)栈空间的开销
class Solution {
private List> ans = new ArrayList<>();
private List arr = new ArrayList<>();
public List> pathSum(TreeNode root, int targetSum) {
if(root==null) return ans;
dfs(root,targetSum);
return ans;
}
private void dfs(TreeNode root,int targetSum){
if(root==null){
return;
}
targetSum -= root.val;
arr.add(root.val);
if(targetSum==0&&root.left==null&&root.right==null){
ans.add(new ArrayList(arr));
}
dfs(root.left,targetSum);
dfs(root.right,targetSum);
arr.remove(arr.size()-1);
}
}
广度优先遍历
- 时间复杂度:O(n)?,n为节点个数
- 空间复杂度:O(n)栈空间的开销