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

最新推荐文章于 2022-04-07 13:16:23 发布

原创最新推荐文章于 2022-04-07 13:16:23 发布 · 872 阅读

0 ·

CC 4.0 BY-SA版权

刷题专栏收录该内容

55 篇文章

订阅专栏

本文介绍了一道经典的二叉树回溯算法题目：寻找所有节点值之和等于指定值的路径。通过递归方法实现，从根节点到叶子节点的路径形成一条完整路径。代码示例使用Java实现。

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

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

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

sum =22
[
   [5,4,11,2],
   [5,8,4,5]
]

一道之前大三的时候面试百度问过的题目。。

一道回溯题，虽然自己写的效率不太好

import java.util.LinkedList;
import java.util.List;

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

    private void dfs(TreeNode root, int sum, List<Integer> temp, List<List<Integer>> res) {
        temp.add(root.val);
        if (sum == root.val && root.left == null && root.right == null) {
            res.add(new LinkedList<>(temp));
            temp.remove(temp.size() - 1);
            return;
        }
        if (root.left != null) dfs(root.left, sum - root.val, temp, res);
        if (root.right != null) dfs(root.right, sum - root.val, temp, res);
        temp.remove(temp.size() - 1);
    }
}