题解 | #二叉树根节点到叶子节点和为指定值的路径#
二叉树根节点到叶子节点和为指定值的路径
http://www.nowcoder.com/practice/840dd2dc4fbd4b2199cd48f2dadf930a
感觉这题跟先序遍历是一个道理,在访问节点值的地方,将值放入到预先创建好的List集合里,然后加上叶子节点的判断条件,符合条件,就将list复制一份放到新创建的List集合中。我自己的思路感觉注意点有两个地方: 第一个是result作为局部变量,每个方法拥有一份,不共享,不能将它定义为成员变量 第二个是把左右子树都遍历完的节点的值从List集合中删除,即list.remove(list.size() - 1);
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<>>
*/
ArrayList<ArrayList<Integer>> finalList = new ArrayList<>();
ArrayList<Integer> list = new ArrayList<>();
public ArrayList<ArrayList<Integer>> pathSum (TreeNode root, int sum) {
// write code here
calculateNode(root, sum, 0);
return finalList;
}
public void calculateNode(TreeNode root, int sum, int result){
if(root != null){
list.add(root.val);
if(root.left == null && root.right == null && result + root.val == sum){
finalList.add(new ArrayList<>(list));
}
calculateNode(root.left, sum, result + root.val);
calculateNode(root.right, sum, result + root.val);
list.remove(list.size() - 1);//移除完全遍历过的节点的值
}
}
}