《剑指offer》 二叉树中和为某一值的路径

题目:

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

思路:

利用先序遍历的递归思想,先遍历树的根节点,然后是左子树,最后是右子树。最后遍历的路径会一直顺着左子树把一条路径走完,每走一个节点,target-该节点的值,一直到路径遍历到叶子节点,判断最后target剩余是不是0,是就把上述路径加到list里,不是就删掉list路径里最后一个点,然后先序遍历会退回到叶子节点的父节点,继续遍历下一个叶子节点(如果有)。随着先序遍历走完整棵树,所有的路径都会加到List里。

1、这道题有点难度,首先记住一句话:在树的前序、中序、后序遍历中,只有前序遍历是首先遍历根结点的(因此需要先遍历根结点的题,就是考查前序遍历)
2、代码注释里写的比较清楚了,首先就是要把当前结点(首先是根结点)添加到路径里,同时target 减去当前结点的值;然后,如果当前结点为叶节点并且和也达到给定值,就把这个路径添加到列表,否则就一直遍历下去;最后,遍历到叶节点之后,返回上层结点之前,一定要把最后一个结点从路径中删除
3、把符合条件的路径添加到列表中的时候,因为add添加的是引用,如果不是每次都new一个path的话,最终list保存到的只是最后一个path(可以看一下ArrayList的源码)

作者:_minimal
链接:https://www.jianshu.com/p/c661cbe1d421
來源:简书
简书著作权归作者所有,任何形式的转载都请联系作者获得授权并注明出处。

牛客网解法

import java.util.ArrayList;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }

}
*/
public class Solution {
    ArrayList<ArrayList<Integer>> patharr =new ArrayList<ArrayList<Integer>>();
        ArrayList<Integer> path=new ArrayList<Integer>();
         public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
                if(root==null)
                    return patharr;
                path.add(root.val);
                target=target-root.val;
                if(target==0&&root.left==null&&root.right==null)
                    patharr.add(new ArrayList<Integer>(path));
                FindPath(root.left,target);            
                FindPath(root.right,target);
                path.remove(path.size()-1);
                return patharr;
            }   
}

有输入输出的解法+注释

作者:_minimal
链接:https://www.jianshu.com/p/c661cbe1d421
來源:简书
简书著作权归作者所有,任何形式的转载都请联系作者获得授权并注明出处。

import java.util.Scanner;
import java.util.ArrayList;
import java.util.Arrays;

public class Solution {
    ArrayList<Integer> path = new ArrayList<>();
    ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();    
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        if(root == null) 
            return list;       
        //每访问到一个结点的时候,都把当前的结点添加到路径中去,并调整target的值
        path.add(root.val);
        target -= root.val;     
        //已到叶节点并且和为target,则把当前路径添加到输出列表里
        //因为add添加的是引用,如果不new一个的话,最终list保存到的只是最后一个path
        if(target == 0 && root.left == null && root.right == null)
            list.add(new ArrayList<Integer>(path));  
        //否则继续遍历
        FindPath(root.left, target);
        FindPath(root.right, target);    
        //已到叶节点之后会跳过两个递归函数到这里,此时要把最后一个结点从路径中删除,才能返回上层结点
        path.remove(path.size()-1);
        return list;
    }
    public static void main(String[] args) {
        // Scanner input=new Scanner(System.in);
        // String st=input.nextLin e();
        Solution s=new Solution();
        TreeNode root = s.new TreeNode(1);
        TreeNode n1 = s.new TreeNode(1);
        TreeNode n2 = s.new TreeNode(1);
        TreeNode n3 = s.new TreeNode(2);
        TreeNode n4 = s.new TreeNode(2);
        TreeNode n5 = s.new TreeNode(1);
        root.left = n1;
        root.right = n2;
        n2.left = n3;
        n2.right = n4;
        n3.left = n5;   
        int[] number = { 1, 2, 3, 4, 5 };
         ArrayList<ArrayList<Integer>> res1 = new ArrayList<>();
         res1=s.FindPath(root, 4);
        // input.close();
    }

}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值