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

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

(题目来自牛客网)
在这里插入图片描述

思路

参考牛客网上的同学的思路分析:

本题要求找到从根节点到各个叶子节点的每一条路径,并把路径上的所有节点之和与某个数相比较,这符合深度优先遍历(DFS)的特点。
参考了几位同学的做法,主要方法useDFS的思路如下:
一、将传入方法的节点的值加入路径列表path。
二、判断该节点是否有孩子节点。

   若没有孩子,则该节点为叶子节点,表示该路径已走完。比较该节点值和目标数值target是否相同,相同则把路径列表path添加到结果列表list中。
   若有孩子,表示该路径还没走完。开始递归:
              1.左孩子不为空,将左孩子节点以及新的目标数值(target-该节点值)作为参数传入useDFS,实现对左孩子的深度优先遍历。
              2.右孩子不为空,进行相同的操作。 

三、以上几步均完成后,表示本次方法中的节点及其左右孩子都已经被遍历完毕,也就是说经过本节点的路径已经统计完毕。此时对路径列表list进行回溯,移除其最后一个元素,结束方法。

注:将路径列表path添加到结果列表list中时,不要直接list.add(path),这样添加的所有元素都是同一个path的引用,是错的。此处应为 list.add(new ArrayList(path));

代码

import java.util.ArrayList;
public class FindPath {
    ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
    ArrayList<Integer> path = new ArrayList<Integer>();
    public ArrayList<ArrayList<Integer>> findPath(TreeNode root, int target) {
        if (root == null){
            return list;
        }
        path.add(root.val);//因为FindPath中和 下面程序中都进行了判null操作,root绝对不可能为 null
        target -= root.val;
        if (target == 0&& root.left == null && root.right == null){//已经到达叶子节点,并且正好加出了target
            list.add(new ArrayList<Integer>(path));//将该路径加入list结果集中
        }
        findPath(root.left,target);//如果左子树非空,递归左子树
        findPath(root.right,target);//如果右子树非空,递归右子树
        path.remove(path.size() -1); //无论当前路径是否加出了target,必须去掉最后一个,然后返回父节点,去查找另一条路径,最终的path肯定为null
        return list;
    }
        public static void main(String[] args) {
        TreeNode node_10 = new TreeNode(10);
        TreeNode node_5 = new TreeNode(5);
        TreeNode node_12 = new TreeNode(12);
        TreeNode node_4 = new TreeNode(4);
        TreeNode node_7 = new TreeNode(7);

        node_10.left = node_5;
        node_10.right = node_12;
        node_5.left = node_4;
        node_5.right = node_7;
        System.out.println(new FindPath().findPath(node_10, 22));
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值