Advertisement
Guest User

All Subtrees With a Given Sum

a guest
Jan 13th, 2019
234
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Java 0.73 KB | None | 0 0
  1.   private static List<List<Tree<Integer>>> getTreesWithSum(Tree<Integer> current,
  2.                                                              int currentSum, int targetSum,
  3.                                                              List<List<Tree<Integer>>> result) {
  4.         currentSum += current.value;
  5.  
  6.         for (Tree<Integer> child : current.children) {
  7.             currentSum += child.value;
  8.  
  9.             getTreesWithSum(child, 0, targetSum, result);
  10.         }
  11.  
  12.         if (currentSum == targetSum) {
  13.             List<Tree<Integer>> subtree = new ArrayList<>();
  14.  
  15.             subtree.add(current);
  16.  
  17.             subtree.addAll(current.children);
  18.  
  19.             result.add(subtree);
  20.         }
  21.  
  22.         return result;
  23.     }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement