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

来源:互联网 发布:量子纠缠知乎 编辑:程序博客网 时间:2024/06/03 14:41

题目:

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


解题思路:

从根节点为起始节点,每次递归的时候target每次先减去当前节点的指,之后判断是否为叶节点,如果是叶节点则判断target是否为零,为零则找到一条路径。否则没有找到,则继续递归。值得注意的是,递归函数每次自动返回到它的父节点,因此,我们在函数退出之前要在路径上删除当前节点以确保返回父节点时候路径刚好是从根节点到父节点。


代码:

public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {    ArrayList<ArrayList<Integer>> res = new ArrayList<>();    ArrayList<Integer> path = new ArrayList<>();    if (root == null) {        return res;    }    findPathHelper(res, path, root, target);    return res;}private void findPathHelper(ArrayList<ArrayList<Integer>> res, ArrayList<Integer> path, TreeNode root, int target) {    if (root == null) {        return;    }    target -= root.val;    if (root.left == null && root.right == null && target == 0) {        path.add(root.val);        //一定要new一个新的,否则直接加进去        //的话之后递归返回无法找到之前的path        res.add(new ArrayList(path));                path.remove(path.size()-1);    }    path.add(root.val);    findPathHelper(res, path, root.left, target);    findPathHelper(res, path, root.right, target);    path.remove(path.size()-1);}



0 0
原创粉丝点击