【LeetCode】 二叉树中和为某一值的路径Java题解

371 阅读1分钟

这是我参与8月更文挑战的第21天,活动详情查看:8月更文挑战

题目描述

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

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

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

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路分析

  • 今天的每日一题是树和路径考察结合的题目,有难度,需要细细思考,反复练习。
  • 题干指出从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。 因此,这里采用前序遍历的思想遍历得到路径。前序遍历使用递归的代码解决,容易理解。
  • 由于可能有多条路径满足条件,采用回溯的思路解决问题。回溯算法的基本思想是:从一条路往前走,能进则进,不能进则退回来,换一条路再试。

通过代码

/**
 * 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 {
    List<List<Integer>> ans = new ArrayList<>();
    List<Integer> path = new ArrayList<>();

    public List<List<Integer>> pathSum(TreeNode root, int target) {
        dfs(root, target);
        return ans;
    }
    
    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) {
            ans.add(new ArrayList<Integer>(path));
        }

        dfs(root.left, target);
        dfs(root.right, target);
        path.remove(path.size() - 1);

    }
}

image.png

总结

  • 上述代码的时间复杂度是O(n), 空间复杂度是O(n)
  • 坚持算法每日一题,加油!