牛客网刷题-二叉树根节点到叶子节点和为指定值的路径

问题描述

给定一个二叉树和一个值 sum，请找出所有的根节点到叶子节点的节点值之和等于 sum 的路径。

输入描述:
输入一棵二叉树，一个sum和

输出描述:
输出所有满足条件的路径

示例

示例1

输入
给出如下的二叉树，sum=22，

输出
[
[5,4,11,2],
[5,8,9]
]

解决思路

分析

1. 采用深度优先搜索遍历的方式，通过回溯法记录走过的路径，判断每一条路径是否符合条件。

深度优先遍历：从跟节点遍历，延某一条路径直到遍历到叶子节点再返回，继续遍历其他路径
回溯法 ：（简述：试探走每一步，达不到条件退回上一步）
是一种选优搜索法，又称为试探法，按选优条件向前搜索，以达到目标。但当探索到某一步时，发现原先选择并不优或达不到目标，就退回一步重新选择，这种走不通就退回再走的技术为回溯法

方法

1. 采用深度优先搜索遍历的方式，通过回溯法记录走过的路径，判断每一条路径是否符合条件。

递归分析
递归参数：
result：存储所有结果；path：存储当前访问到的路径；root：二叉树；sum：路径和
终止条件：
1、到达叶子节点，没有满足sum
2、到达叶子节点，路径和为sum
递归内容：
1、路径中添加节点
2、递归左节点和右节点
3、回溯点：剔除上一步的结果

代码实现

// 思路1
public class Solution {  
    
    private void dfs (ArrayList<ArrayList<Integer>> result, ArrayList<Integer> path, TreeNode root, int sum) {
        if (root == null) {
            return;
        }

		// 到达叶子节点，路径和为sum
        if (root.left == null && root.right == null) {
            // 到达叶子节点，没有满足sum
            if (sum - root.val == 0) {
                path.add(root.val);
                result.add(new ArrayList<>(path));
                path.remove(path.size() - 1);
            }
            // 到达叶子节点，没有满足sum
            return;
        }
        // 路径中添加节点
        path.add(root.val);
        // 递归左节点和右节点
        dfs(result, path, root.left, sum - root.val);
        dfs(result, path, root.right, sum - root.val);
        // 回溯点
        path.remove(path.size() - 1);
    }
}

时间复杂度分析：
深度优先搜索遍历会遍历所有的节点，所以时间复杂度为 O(n)

空间复杂度分析：
因为会申请额外的空间path作为回溯法存储路径的结构，path最长路径为n（即树为链表结构），所以空间复杂度为 O(n)

小伙伴如果想测试的话，可以直接到牛客网这个链接做测试

二叉树根节点到叶子节点和为指定值的路径-牛客网