【其他】从任意结点出发到叶结点,寻找和等于目标值的序列

package top.actim.test09;

import java.util.ArrayList;

public class Test02 {
	
	
	/**
	 *    4
	 *  9   0
	 * 5 1
	 */
	public static void main(String[] args) {
		TreeNode root = new TreeNode(4) ;
        TreeNode node = new TreeNode(9) ;
        TreeNode node1 = new TreeNode(0) ;
        TreeNode node2 = new TreeNode(5) ;
        TreeNode node3 = new TreeNode(1) ;
        root.left = node ;
        root.right = node1 ;
        node.left = node2 ;
        node.right = node3 ;
		
        
        dfs(root, 1, new ArrayList<Integer>());
        System.out.println(ans);ans.clear();
        dfs(root, 10, new ArrayList<Integer>());
        System.out.println(ans);ans.clear();
        dfs(root, 14, new ArrayList<Integer>());
        System.out.println(ans);ans.clear();
        dfs(root, 18, new ArrayList<Integer>());
        System.out.println(ans);ans.clear();
        dfs(root, 4, new ArrayList<Integer>());
        System.out.println(ans);ans.clear();
	}

	static ArrayList<ArrayList<Integer>> ans = new ArrayList<ArrayList<Integer>>();

	static void dfs(TreeNode node, int sum, ArrayList<Integer> temp) {
		if (node == null)
			return;
		if (node.left == null && node.right == null) {
//			System.out.println(temp);
			ArrayList<Integer> tt = new ArrayList<Integer>();
			tt.add(node.val);
			int count = node.val;
			for (int i = temp.size() - 1; i >= 0; i--) {
				if(count == sum) {
					ans.add(new ArrayList<Integer>(tt));
				}
				tt.add(0, temp.get(i));
				count += temp.get(i);
			}
			if(count == sum) {
				ans.add(new ArrayList<Integer>(tt));
			}
		}
		
		temp.add(node.val);
		dfs(node.left, sum, temp);
		dfs(node.right, sum, temp);
		temp.remove(temp.size() - 1);
	}

}

class TreeNode {
    public int val;
    public TreeNode left;
    public TreeNode right;
    TreeNode(int x) { val = x; }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 12
    评论
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值