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

AsiaSun.

已于 2024-01-04 08:58:00 修改

阅读量137

点赞数

分类专栏：剑指Offer 文章标签：数据结构

于 2021-11-04 23:29:14 首次发布

本文链接：https://blog.csdn.net/IOSSHAN/article/details/121153923

版权

剑指Offer 专栏收录该内容

35 篇文章 0 订阅

订阅专栏

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

给你二叉树的根节点 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

思路

先序遍历
在先序遍历中，记录从根节点到当前节点的路径
如果当前节点是叶子节点，且
路径上的值等于目标值，则该路径就是所求的一条路径

/**
 * Definition for a binary tree node.
 * public 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 {
    //结果二维数组
    LinkedList<List<Integer>> resoultArray = new LinkedList<List<Integer>>();

    //路径-维数组
    LinkedList<Integer> pathArray = new LinkedList<Integer> ();

    public List<List<Integer>> pathSum(TreeNode root, int target) {
        rec(root, target);
        return resoultArray;
    }

    public void rec(TreeNode root, int target) {
        //如果为null，则返回
        if (root == null) return;
        
        //将值加入数组
        pathArray.add(root.val);

        //目标值变小
        target = target - root.val;
        
        //如果是叶子结点，且target == 0，则满足条件
        if (target == 0 && root.left == null && root.right == null) {
            resoultArray.add(new ArrayList<Integer>(pathArray));
        }
        
        
        rec(root.left, target);
        rec(root.right, target);
        //每次需要把最后一个叶子节点去掉
        pathArray.removeLast();

    }
}