Updated ArrayStack

William Fiset · William Fiset · commit aa65bb692850 · 2020-06-04T19:50:43.000-07:00
diff --git a/src/main/java/com/williamfiset/algorithms/datastructures/stack/ArrayStack.java b/src/main/java/com/williamfiset/algorithms/datastructures/stack/ArrayStack.java
@@ -5,13 +5,12 @@
 
 /** @author liujingkun */
 public class ArrayStack<T> implements Stack<T> {
+  private int size;
   private int capacity;
   private Object[] data;
-  private int size;
 
   public ArrayStack() {
     capacity = 16;
-    size = 0;
     data = new Object[capacity];
   }
 
@@ -28,17 +27,14 @@ public boolean isEmpty() {
   @Override
   public void push(T elem) {
     if (size == capacity) {
-      adjustCap();
+      increaseCapacity();
     }
     data[size++] = elem;
   }
 
-  // adjust the capacity to store more element
-  private void adjustCap() {
-    if (capacity == Integer.MAX_VALUE) {
-      throw new OutOfMemoryError();
-    }
-    capacity += capacity;
+  // Increase the capacity to store more elements.
+  private void increaseCapacity() {
+    capacity *= 2;
     data = Arrays.copyOf(data, capacity);
   }
 
