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

题目描述

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

这里写图片描述

如图,输入上图的二叉树和整数 22,则打印出两条路径,第一条路径:10, 12; 第二条路径为:10, 5, 7.

由于路径是从根节点出发到叶节点,也就是说路径总是以根节点为起始点,因此首先需要遍历根节点。在树的前序,中序,后序三种遍历方式中,只有前序遍历是首先访问根节点。

遍历过程如下表:
这里写图片描述

分析上述过程,我们找到了一些规律:当用前序遍历的方式访问到某一节点时,我们把这个节点添加到路径上,并累加该节点的值,如果该节点为叶子节点并且路径中节点值的和刚好等于输入的整数,则当前的路径符合要求,我们把它打印出来。如果当前节点不是叶节点,则继续访问它的子节点。当前节点访问结束后,递归函数将自动回到它的父节点。因此我们在函数退出之前要在路径上删除当前节点,并减去当前节点的值,以确保返回父节点时路径刚好是从根节点到父节点的路径。不难看出保存路径的数据结构实际上是一个栈,因为路径要与递归调用状态一致,而递归调用的本质就是一个压栈和出栈的过程。

 public class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

        public TreeNode(int val) {
            this.val = val;

        }

    }
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        if(root == null){
            return new ArrayList();
        }
        ArrayList<ArrayList<Integer>> array = new ArrayList<ArrayList<Integer>>();
        int current = 0 ;
        Stack<Integer> path = new Stack<Integer>();
        // ArrayList<Integer> array = new ArrayList<Integer>();
        FindPath(root , target , path , current , array);
        return array;
    }
    void FindPath(TreeNode root , int target ,Stack<Integer> path ,  int current ,ArrayList<ArrayList<Integer>> array){
        current += root.val;
        path.push(root.val);
        boolean isLeaf =  (( root.left == null) && ( root.right == null));
        if(current == target && isLeaf){
            array.add(new ArrayList<Integer>(path));
        }
        if(root.left != null){
            FindPath(root.left , target , path , current, array);
        }
        if(root.right != null){
            FindPath(root.right , target , path , current , array);
        }
        current -=root.val;
        path.pop();
    }

查看文章:https://my.oschina.net/u/2260265/blog/345667

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值