面试题34. 二叉树中和为某一值的路径

最新推荐文章于 2021-07-08 17:22:21 发布

@渐行渐远

最新推荐文章于 2021-07-08 17:22:21 发布

阅读量97

点赞数

分类专栏：剑指Offer 面试文章标签：二叉树链表 leetcode java

本文链接：https://blog.csdn.net/qq_44361838/article/details/115490016

版权

面试同时被 2 个专栏收录

79 篇文章 0 订阅

订阅专栏

剑指Offer

68 篇文章 0 订阅

订阅专栏

该博客主要介绍了如何使用回溯算法解决二叉树中节点值之和等于目标值的路径问题。通过先序遍历，对每个节点进行判断，如果路径和等于目标值并且当前节点为叶子节点，则将路径加入结果集。在回溯过程中，使用LinkedList的removeLast()函数确保路径正确。此外，注意在记录路径时要创建路径副本，避免路径改变影响已存储的结果。

摘要由CSDN通过智能技术生成

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

示例:
给定如下二叉树，以及目标和 target = 22，

          5
         / \
        4   8
       /   / \
      11  13  4
     /  \    / \
    7    2  5   1

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

思路：

这个题目使用回溯的思想。
res 保存结果集，path保存路径。
在途中我们使用先序遍历，然后进行遍历，当到达根节点不满足条件的时候，我们调用LinkedList的removelast函数，来删除最后一个节点，保证回溯的path是正确的。
注意：这里需要注意的一个点是，当我们找到一个满足条件的path之后，我们本来会想当然的使用res.add(path)，把这个path加入到list中，但是我们忽视了一点，值得注意的是，记录路径时若直接执行 res.add(path) ，则是将 path 对象加入了 res ；后续 path 改变时， res 中的 path 对象也会随之改变。
所以我们需要在这里新建一个LinkedList，复制一个副本。

复杂度：

时间：O(N) 遍历所有的二叉树节点。
空间：O(N) 当二叉树退化为链表的时候，我们的path 最大需要N个空间。

// class TreeNode {
//       int val;
//       TreeNode left;
//       TreeNode right;
//       TreeNode() {}
//      TreeNode(int val) { this.val = val; }
//       TreeNode(int val, TreeNode left, TreeNode right) {
//          this.val = val;
//           this.left = left;
//           this.right = right;
//       }
//   }

class Solution {
    List<List<Integer>> res=new ArrayList<>();
    LinkedList<Integer> path=new LinkedList<>(); 
    public List<List<Integer>> pathSum(TreeNode root, int target) {
       
       dfs(root,target);
       return res;
    }
    public void dfs(TreeNode root,int target){
        if(root==null)
            return;
        path.add(root.val);
        target-=root.val;
        if(target==0&&root.left==null&&root.right==null)
            res.add(new LinkedList<>(path));
        dfs(root.left,target);
        dfs(root.right,target);
        path.removeLast();    
    }
}