Bootstrap

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

描述

输入一颗二叉树的根节点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;
    }
}

 

;