JZ34 二叉树中和为某一值的路径(二)

最新推荐文章于 2022-09-10 09:23:24 发布

自律信仰

最新推荐文章于 2022-09-10 09:23:24 发布

阅读量290

点赞数

分类专栏：剑指 # 算法文章标签：深度优先算法

本文链接：https://blog.csdn.net/hza163/article/details/124764729

版权

算法同时被 2 个专栏收录

115 篇文章 0 订阅

订阅专栏

剑指

28 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何在给定的二叉树中寻找节点值之和等于特定数值的路径。作者提供了两种Java实现，分别使用ArrayList和LinkedList，强调了在处理路径时需要注意的引用传递问题，并给出了详细的代码示例。此外，还提到可以使用栈来解决这个问题。

摘要由CSDN通过智能技术生成

描述

输入一颗二叉树的根节点root和一个整数expectNumber，找出二叉树中结点值的和为expectNumber的所有路径。

1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点

2.叶子节点是指没有子节点的节点

3.路径只能从父节点到子节点，不能从子节点到父节点

4.总节点数目为n

如二叉树root为{10,5,12,4,7},expectNumber为22

则合法路径有[[10,5,7],[10,12]]

数据范围:

树中节点总数在范围 [0, 5000] 内

-1000 <= 节点值 <= 1000

-1000 <= expectNumber <= 1000

示例1

输入：

{10,5,12,4,7},22

返回值：

[[10,5,7],[10,12]]

说明：

返回[[10,12],[10,5,7]]也是对的

示例2

输入：

{10,5,12,4,7},15

返回值：

[]

示例3

输入：

{2,3},0

返回值：

[]

示例4

输入：

{1,3,4},7

返回值：

[]

算法很简单，一个全局变量+DFS即可

初用java卡在语法上，java ArrayList传递的是引用（地址），为了传值得复制一份或者DFS结尾删除，但是ArrayList又不支持删除最后一个得改用LinkedList

代码1：

    ArrayList<ArrayList<Integer>> ans=new ArrayList<>();
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int expectNumber) {
        ArrayList<Integer> path = new ArrayList<>();
        DFS(root,expectNumber,path);
        return ans;
    }

    public void DFS(TreeNode root, int expectNumber,ArrayList<Integer> path){
        if(root==null) return;
        path.add(root.val);
        if(root.left==null&&root.right==null&&expectNumber-root.val==0) ans.add(path);

        //暂时不会复制API 赶紧自己实现，也不难嘛   不过就这小问题为了找API竟然卡了好久
        ArrayList<Integer> path2=new ArrayList<>();
        for (Integer i : path)  path2.add(i);

        DFS(root.left,expectNumber-root.val,path);
        DFS(root.right,expectNumber-root.val,path2);
    }

代码2：

    ArrayList<ArrayList<Integer>> ans=new ArrayList<>();
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int expectNumber) {
        LinkedList<Integer> path = new LinkedList<>();//想要删除得换成LinkedList
        DFS(root,expectNumber,path);
        return ans;
    }

    public void DFS(TreeNode root, int expectNumber,LinkedList<Integer> path){
        if(root==null) return;
        path.add(root.val);
        if(root.left==null&&root.right==null&&expectNumber-root.val==0)
            ans.add(new ArrayList<>(path));

        DFS(root.left,expectNumber-root.val,path);
        DFS(root.right,expectNumber-root.val,path);

        path.removeLast();//想要删除得换成LinkedList  ArrayList没法删除最后一个
    }

题解代码:

学会变通用栈也行，那就用呗

public class Solution {
    private ArrayList<ArrayList<Integer>> paths = new ArrayList<>();
    private Stack<Integer> path = new Stack<>();

    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        if(root ==null) return paths;
        path.push(root.val);
        target -= root.val;

        if(target == 0 && root.left == null && root.right ==null){
            paths.add(new ArrayList<Integer>(path));
        }

        FindPath(root.left,target);
        FindPath(root.right,target);
        path.pop();

        return paths;
    }
}