diff --git a/src/Foo.java b/src/Foo.java index 763bf78..4952d5b 100644 --- a/src/Foo.java +++ b/src/Foo.java @@ -3,21 +3,22 @@ public class Foo { - private List list; + private List numbers; public void add(int i) { - if(list == null) { - list = new ArrayList<>(); + if(numbers == null) { + numbers = new ArrayList<>(); } - list.add(i); + numbers.add(i); } public void remove(int i) { - if(list.contains(i)) - list.remove(i); + if(numbers != null) + numbers.remove(i); } public int sum() { - return list.stream().mapToInt(Integer::intValue).sum(); + return (numbers == null) ? 0 : + numbers.stream().mapToInt(Integer::intValue).sum(); } } \ No newline at end of file