exo2: simplify iterator loops.

[Project_POO.git] / exo2 / ListExtension.java

diff --git a/exo2/ListExtension.java b/exo2/ListExtension.java
index bba85d2fb1c60307cdf0d25333d18f6de501fb5d..3ba71ac607d993661cb26597093ae35c668ce024 100644 (file)
--- a/exo2/ListExtension.java
+++ b/exo2/ListExtension.java
@@ -41,24 +41,22 @@ public class ListExtension<E> {
         int mid = list.size() / 2;
         ListIterator<E> iter = list.listIterator();
         int i = 0;
-        // go to the element at mid index
-        while (iter.hasNext() && i < mid) {
+        // go to the element at mid + 1 index
+        while (iter.hasNext() && i <= mid) {
             iter.next();
             i++;
         }
-        iter.next(); // Insert before mid + 1
         iter.add(value);
     }
 
     public void addNEMiddle(int Nelements) {
         ListIterator<E> iter = list.listIterator();
         int i = 0;
-        // go to the element at the middle index
-        while (iter.hasNext() && i < list.size() / 2) {
+        // go to the element at the middle + 1 index
+        while (iter.hasNext() && i <= list.size() / 2) {
             iter.next();
             i++;
         }
-        iter.next(); // Insert before mid + 1
         for (int j = 0; j < Nelements; j++) {
             iter.add((E)new Object());
         }