Path Sum II 二叉树路径和,打印所有路径 @LeetCode

题目:

给定一个二叉树和一个值sum,求这个二叉树是否存在一条从root到叶子节点的路径以满足路径和和sum相等。

并打印出所有路径

思路:

经典递归回溯!



package Level2;

import java.util.ArrayList;

import Utility.TreeNode;

/**
 * Path Sum II  
 * 
 * Given a binary tree and a sum, find all root-to-leaf paths where each path's sum equals the given sum.

For example:
Given the below binary tree and sum = 22,
              5
             / \
            4    8
           /     / \
          11  13  4
         /  \      / \
        7    2   5   1
return
[
   [5,4,11,2],
   [5,8,4,5]
]
 */


public class S113 {

	public static void main(String[] args) {
		TreeNode root = new TreeNode(5);
		TreeNode n1 = new TreeNode(4);
		TreeNode n2 = new TreeNode(8);
		root.left = n1;
		root.right = n2;
		TreeNode n3 = new TreeNode(11);
		TreeNode n4 = new TreeNode(13);
		TreeNode n5 = new TreeNode(4);
		n1.left = n3;
		n2.left = n4;
		n2.right = n5;
		TreeNode n6 = new TreeNode(7);
		TreeNode n7 = new TreeNode(2);
		n3.left = n6;
		n3.right = n7;
		TreeNode n8 = new TreeNode(5);
		TreeNode n9 = new TreeNode(1);
		n5.left = n8;
		n5.right = n9;
		
		ArrayList<ArrayList<Integer>> list = pathSum(root, 22);
		System.out.println(list);
	}
	
	public static ArrayList<ArrayList<Integer>> pathSum(TreeNode root, int sum) {
		ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
		ArrayList<Integer> l =new ArrayList<Integer>();
		dfs(root, sum, list, l);
		return list;
    }
	
	// 经典递归回溯
	private static void dfs(TreeNode root, int sum, ArrayList<ArrayList<Integer>> list, ArrayList<Integer> l){
		if(root == null){
			return;
		}
	
		// 找到最后一个合适的叶子节点
		if(root.val==sum && root.left==null && root.right==null){
			l.add(root.val);
			
			// 注意!在把结果加入结果集时,必须深拷贝一份!
			// 否则被加进去的集合之后可能会变动
			ArrayList<Integer> clone = new ArrayList<Integer>(l);
			list.add(clone);
			
			l.remove(l.size()-1);		// 恢复全局变量//回溯部分
			return;
		}
	
		l.add(root.val);
		dfs(root.left, sum-root.val, list, l);
		dfs(root.right, sum-root.val, list, l);
		l.remove(l.size()-1);		// 恢复全局变量//回溯部分
	}
}


注意递归的base case不光要考虑到root为null情况,而且要考虑到叶子节点的情况!

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        List<List<Integer>> ret = new ArrayList<List<Integer>>();
        List<Integer> list = new ArrayList<Integer>();
        rec(ret, list, root, sum);
        return ret;
    }
    
    public void rec(List<List<Integer>> ret, List<Integer> list, TreeNode root, int sum) {
        if(root == null) {
            return;
        }
        if(root.left==null && root.right==null && sum == root.val){
            list.add(root.val);
            ret.add(new ArrayList<Integer>(list));
            list.remove(list.size()-1);
            return;
        }
        
        list.add(root.val);
        rec(ret, list, root.left, sum-root.val);
        rec(ret, list, root.right, sum-root.val);
        list.remove(list.size()-1);
    }
}




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值