[剑指OFFER]之树：JZ34 二叉树中和为某一值的路径(二)

最新推荐文章于 2025-11-26 15:59:01 发布

原创最新推荐文章于 2025-11-26 15:59:01 发布 · 211 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构 #后端 #算法 #开发语言 #java

数据结构与算法专栏收录该内容

42 篇文章

订阅专栏

文章目录

@[TOC]
1.题目
2.解法
2.1 树的结构
2.2 解法1
2.3 解法2

总结

1.题目

输入一颗二叉树的根节点root和一个整数expectNumber，找出二叉树中结点值的和为expectNumber的所有路径。
1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点
2.叶子节点是指没有子节点的节点
3.路径只能从父节点到子节点，不能从子节点到父节点
4.总节点数目为n

2.解法

2.1 树的结构

 class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;
        public TreeNode(int val) {
            this.val = val;
        }
    }

2.2 解法1

这道题并没有固定“从根节点开始”这一条件，因此需要考虑每个节点都当做根节点开始递归。也就是说每个节点都需要从头开始捋一遍。

    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
        ArrayList<ArrayList<Integer>> ans = new ArrayList<>();
        if (root == null) return ans;
        ArrayList<Integer> path = new ArrayList<>();
        helper(ans, path, root, target);
        return ans;
    }

    public void helper(ArrayList<ArrayList<Integer>> ans, ArrayList<Integer> path, TreeNode node, int target) {
        if (node == null) return;

        path.add(node.val);

        if (node.left == null && node.right == null && target == node.val) {
            ans.add(new ArrayList<>(path));
            return;
        }


        helper(ans, new ArrayList<>(path), node.left, target - node.val);
        helper(ans, new ArrayList<>(path), node.right, target - node.val);
    }

2.3 解法2

使用栈来模拟，类似与回溯算法。
跟之前的优势在于，不需要每次传入一个新的ArrayList<Integer> path来存储路径，而是可以直接复用Stack<Integer> path。

   public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
        ArrayList<ArrayList<Integer>> ans = new ArrayList<>();
        if (root == null) return ans;
        Stack<Integer> path = new Stack<>();
        helper(ans, path, root, target);
        return ans;
    }

    public void helper(ArrayList<ArrayList<Integer>> ans, Stack<Integer> path, TreeNode node, int target) {

        if (node == null) return;
        if (node.left == null && node.right == null && target == node.val) {
            path.push(node.val);
            ans.add(new ArrayList<>(path));
            return;
        }

        path.push(node.val);
        helper(ans, path, node.left, target - node.val);
        path.pop();

        path.push(node.val);
        helper(ans, path, node.right, target - node.val);
        path.pop();
    }