Problem:
Given a binary tree and a sum, determine if the tree has a root-to-leaf path such that adding up all the values along the path equals the given sum.
Explanation:
给出一棵树和一个数值,找出从根结点到叶子结点的n条路径,让路径上数值总和等于给定的数。
My Thinking:
My Solution:
Optimum Thinking:
在Path Sum I的基础上添加两个list,result存放最后的结果,currentresult存放从根结点到当前结点走过的所有结点,直到叶节点,判断是否等于0,若等于0则将currentresult返回,并将其最后一个元素移除,也就是返回到上一个结点的地方。
Optimum Solution:
public List<List<Integer>> pathSum(TreeNode root, int sum){
List<List<Integer>> result = new LinkedList<List<Integer>>();
List<Integer> currentResult = new LinkedList<Integer>();
pathSum(root,sum,currentResult,result);
return result;
}
public void pathSum(TreeNode root, int sum, List<Integer> currentResult,
List<List<Integer>> result) {
if (root == null)
return;
currentResult.add(new Integer(root.val));
if (root.left == null && root.right == null && sum == root.val) {
result.add(new LinkedList(currentResult));//因为currentresult会一直改变,因此需要复制一份。
currentResult.remove(currentResult.size() - 1);//这里要删除最后的元素,返回到上一个结点
return;
} else {
pathSum(root.left, sum - root.val, currentResult, result);
pathSum(root.right, sum - root.val, currentResult, result);
}
currentResult.remove(currentResult.size() - 1);
}