diff --git a/ArrayList.py b/ArrayList.py index 607865d..caf9b41 100644 --- a/ArrayList.py +++ b/ArrayList.py @@ -27,6 +27,7 @@ def add_at(self, index, item): if index < 0 or index >= self.size: raise Exception("Index out of range") self.items.insert(index, item) + self.count += 1 def add_first(self, item): self.add_at(0, item) @@ -38,6 +39,7 @@ def remove_at(self, index): if index < 0 or index >= self.size: raise Exception("Index out of range") del self.items[index] + self.count -= 1 def remove_first(self): self.remove_at(0)