24二叉树中的和为某值的路径

▼魔方 西西 提交于 2020-02-19 07:13:52

题目描述

输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)

思路分析

递归遍历树, 把结点加入路径。若该结点是叶子结点且当前路径和等于期待和,则将此路径加入返回的lists中,否则退回父结点。在每一轮递归返回到父结点时,当前路径也要回溯一个结点值。

代码实现

 public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
        ArrayList<ArrayList<Integer>> arrayLists = new ArrayList<>();
        if (root == null || root.val > target) {
            return arrayLists;
        }
        arrayLists = process(root, target, arrayLists, new ArrayList<Integer>());
        return arrayLists;
    }

    public ArrayList<ArrayList<Integer>> process(TreeNode root, int target, ArrayList<ArrayList<Integer>> arrayLists, ArrayList<Integer> list) {
        if(root==null){
            return arrayLists;
        }
        target -= root.val;
        list.add(root.val);
        if (target == 0 && root.right == null && root.left == null) {
            arrayLists.add(new ArrayList<>(list));
        }
        process(root.left, target, arrayLists, list);
        process(root.right, target, arrayLists, list);
        list.remove(list.size() - 1);
        return arrayLists;
    }
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!