二叉树和为某一值的路径

 

 (1) DFS

    List<List<Integer>> res = new LinkedList<List<Integer>>();
    Deque<Integer> path = new LinkedList<>();
    public List<List<Integer>> pathSum(TreeNode root, int target) {
        dfs(root, target);
        return res;
    }
    public void dfs(TreeNode root, int target){
        if(root == null) return;
        path.addLast(root.val);
        target -= root.val;
        if(root.left == null && root.right == null && target ==0){
            res.add(new LinkedList<Integer>(path) );
        }
        dfs(root.left, target);
        dfs(root.right, target);
        path.pollLast();
    }

 (2)BFS +哈希表记录每个节点的父节点。

List<List<Integer>> res = new LinkedList<List<Integer>>();
    Map<TreeNode, TreeNode> map = new HashMap<TreeNode,TreeNode>();
    public List<List<Integer>> pathSum(TreeNode root, int target) {
        if(root == null ) return res;
        Queue<TreeNode> nodeQueue = new LinkedList<>();
        Queue<Integer> sumQueue = new LinkedList<>();
        nodeQueue.offer(root);
        sumQueue.offer(0);
        while(!nodeQueue.isEmpty()){
            TreeNode node = nodeQueue.poll();
            int tmp = sumQueue.poll() + node.val;
            if(node.left == null && node.right == null && tmp == target){
                getPath(node);
            }
            if( node.left != null){
                map.put(node.left, node);
                nodeQueue.offer(node.left);
                sumQueue.offer(tmp);
            }
            if( node.right != null){
                map.put(node.right, node);
                nodeQueue.offer(node.right);
                sumQueue.offer(tmp);
            }
        }
       return res;
    }

    private void getPath(TreeNode node) {
        LinkedList<Integer> list = new LinkedList<>();
        while(node != null){
            list.add(0, node.val);
            node = map.get(node);
        }
        res.add(list);
    }

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值