(Java）leetcode-113 Path Sum II（路径总和 II）

最新推荐文章于 2022-07-07 21:50:23 发布

Mr.Bean-Pig

最新推荐文章于 2022-07-07 21:50:23 发布

阅读量238

点赞数

分类专栏：算法题解文章标签：二叉树 java 数据结构算法 leetcode

本文链接：https://blog.csdn.net/z714405489/article/details/106952960

版权

算法题解专栏收录该内容

140 篇文章 0 订阅

订阅专栏

题目描述

给定一个二叉树和一个目标和，找到所有从根节点到叶子节点路径总和等于给定目标和的路径。

说明: 叶子节点是指没有子节点的节点。

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

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

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

思路

先序遍历并且更新经过的路径和应该难度不大，这题的难点在于路径的记录。

这里用一个全局的LinkedList path随时更新遍历经过的路径，关键步骤就是往上层回溯时，删除path的最后一个记录。

当遍历至叶子节点时，如果路径和与目标相等，则把此时的路径path添加到结果集中（这里要注意得new一个添加进去，否则后面path的更新会影响res中的结果，因为LinkedList是引用数据类型）。

时间复杂度：O(N)
空间复杂度：O(N)

代码

class Solution {
	List<List<Integer>> res = new LinkedList<>();
	List<Integer> path = new LinkedList<>();
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
    	if (root == null) return res;
    	dfs(root, 0, sum);
    	return res;
    }

    private void dfs(TreeNode root, int sum, int target) {
    	if (root == null) return;
    	path.add(root.val);
    	sum += root.val;
    	// 叶子节点
    	if (root.left == root.right) {
    		if (sum == target) res.add(new LinkedList<>(path));  			
    	}
    	dfs(root.left, sum, target);
    	dfs(root.right, sum, target);
    	// 回溯
    	path.remove(path.size()-1);
    }
}