algorithms lesson #6 update

Author: biblelamp

BerlinTelran/Algorithms/src/lesson6/DynamicArray.java

@@ -23,26 +23,29 @@ public int get(int idx) {
     }
 
     public void add(int value) {
-        if (size < array.length) {
-            array[size] = value;
-            size++;
-        } else {
-            int[] extArray = new int[array.length * 2];
-            System.arraycopy(array, 0, extArray, 0, array.length);
-            array = extArray;
-            array[size] = value;
-            size++;
-        }
+        expandArrayIfNeed();
+        array[size] = value;
+        size++;
     }
 
     public void add(int idx, int value) {
-        // TODO you have to control array's capacity
+        expandArrayIfNeed();
         System.arraycopy(array, idx, array, idx + 1, array.length - idx - 1);
         array[idx] = value;
         size++;
     }
 
+    private void expandArrayIfNeed() {
+        if (size == array.length) {
+            int[] extArray = new int[array.length * 2];
+            System.arraycopy(array, 0, extArray, 0, array.length);
+            array = extArray;
+            //System.out.println("Array size *2 = " + array.length);
+        }
+    }
+
     public void remove(int idx) {
+        // TODO check size array
         System.arraycopy(array, idx + 1, array, idx, array.length - idx - 1);
         size--;
     }

BerlinTelran/Algorithms/src/lesson6/Lesson6.java

@@ -15,12 +15,18 @@ public static void main(String[] args) {
         da.add(12);
         da.add(3);
         da.add(-1);
-        da.add(7);
-        da.add(13);
         System.out.println(da);
+        //da.add(7);
+        //da.add(13);
+        //System.out.println(da);
         da.add(2, -5);
         System.out.println(da);
-        da.remove(3);
+        //da.remove(3);
+        da.add(8);
+        da.add(4);
+        da.add(12);
+        da.add(3);
+        da.add(-1);
         System.out.println(da);
     }
 }