您的位置:首页 > 其它

leetcode:Path Sum II

2014-06-11 22:01 225 查看
Path Sum I 的变形,只是在PathSun1的情况要求出所有可能的情况

/**
* Definition for binary tree
* public class TreeNode {
*     int val;
*     TreeNode left;
*     TreeNode right;
*     TreeNode(int x) { val = x; }
* }
*/
public class Solution {

static int ans = 0;
static ArrayList<Integer> stack = new ArrayList<Integer>();
static ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();

public ArrayList<ArrayList<Integer>> pathSum(TreeNode root, int sum) {
ans = 0;
stack.clear();
list.clear();
traverl(root, sum);
return list;
}

public static boolean isLeaf(TreeNode root){
return root.left == null && root.right == null;
}
static void traverl(TreeNode root, int sum){
if(root != null){
stack.add(root.val);
ans += root.val;
if(isLeaf(root) && ans == sum){
ArrayList<Integer> t = new ArrayList<Integer>();
for(int i = 0; i < stack.size();++i){
t.add(stack.get(i));
}
list.add(t);
}
traverl(root.left, sum);
traverl(root.right, sum);
stack.remove(stack.size() - 1);
ans -= root.val;
}
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: