剑指 Offer 34. 二叉树中和为某一值的路径-DFS

Posted 2022-05-02 hequnwang10

一、题目描述

给你二叉树的根节点 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
输出：[]

二、解题

DFS

class Solution 
    public List<List<Integer>> pathSum(TreeNode root, int target) 
        List<List<Integer>> res = new ArrayList<>();
        List<Integer> per = new ArrayList<>();
        if(root == null)
            return res;
        
        dfs(res,per,root,target);
        return res;
    
    public void dfs(List<List<Integer>> res,List<Integer> per,TreeNode root, int target)
        if(root == null)
            return;
        
        per.add(root.val);
        target -= root.val;
        if(root.left == null && root.right == null && target == 0)
            res.add(new ArrayList<>(per));
        
        dfs(res,per,root.left,target);
        dfs(res,per,root.right,target);
        per.remove(per.size()-1);
    

时间复杂度：O(n^2)；

空间复杂度：O(n)。