[算法练习及思路-leetcode剑指offer(Java解法)]No34.二叉树中和为某一值的路径

题号:剑指 Offer 34

题目名:二叉树中和为某一值的路径

原题URL:https://leetcode-cn.com/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof/

题目描述

输入一棵二叉树和一个整数，打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

示例

示例 1：

给定如下二叉树，以及目标和 sum = 22，

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1
返回:

[
   [5,4,11,2],
   [5,8,4,5]
]

限制

• 节点总数 <= 10000

思路

1.使用dfs进行回溯剪枝

2.每次遍历一个结点就将值在sum基础上减

3.当sum==0的时候，就可以输出结果，将当前集合加入到返回集合中

4.注意每次进行递归的时候需要进行剪枝

解题代码

public class Solution {
    List<List<Integer>> res = new ArrayList<>();
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        if(root==null) return res;
        List<Integer> list = new ArrayList<>();
        findPath(root,list,sum);
        return res;
    }

    private void findPath(TreeNode root, List<Integer> list, int sum) {
        if(root==null) return;
        //将当前节点的值加入sum
        list.add(root.val);
        //如果当前节点是根节点
        if(root.left==null&&root.right==null) {
            if(sum-root.val==0) {
                res.add(new ArrayList<>(list));
            }
            return;
        }
        //遍历左子树
        findPath(root.left,list,sum-root.val);
        //回溯
        if(root.left!=null) list.remove(list.size()-1);
        findPath(root.right,list,sum-root.val);
        if(root.right!=null) list.remove(list.size()-1);
    }
}