8.二叉树根节点到叶子节点和为指定值的路径

题目描述

给定一个二叉树和一个值\ sum sum,请找出所有的根节点到叶子节点的节点值之和等于\ sum sum 的路径,
例如:
给出如下的二叉树,\ sum=22 sum=22,

返回
[
[5,4,11,2],
[5,8,9]
]

示例1

输入

{1,2},1

返回值

[]
示例2

输入

{1,2},3
 

返回值

[[1,2]]

代码实现

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @param sum int整型 
     * @return int整型ArrayList<ArrayList<>>
     */
private ArrayList<ArrayList<Integer>> listAll = new ArrayList<>();
private ArrayList<Integer> list = new ArrayList<>();
public ArrayList<ArrayList<Integer>> pathSum(TreeNode root, int target) {
	if (root == null) {
		return listAll;
	}
	list.add(root.val);
	target -= root.val;
	if (target == 0 && root.left == null && root.right == null) {
		listAll.add(new ArrayList<>(list));
	}
	pathSum(root.left, target);
	pathSum(root.right, target);
	// 回退
	list.remove(list.size() - 1);
	return listAll;
}
}

猜你喜欢

转载自blog.csdn.net/xiao__jia__jia/article/details/113480699