TP2: coding style fixlet.

[TP_POO.git] / TP2 / Liste.java

diff --git a/TP2/Liste.java b/TP2/Liste.java
index 201d8750af17c7f0110e8bc852811ed446fea0c7..344a2d3325f134a76399c72892b4dbc996768f18 100644 (file)
--- a/TP2/Liste.java
+++ b/TP2/Liste.java
@@ -143,19 +143,20 @@ public class Liste extends Structure {
             return;
         } else if (nElements == 0) {
             return;
-        } else if (headNode.getNext() == null) {
+        } else if (headNode != null && headNode.getNext() == null) {
             headNode = null;
         } else {
             // We have at least 2 nodes in the linked list
-            IntNode nodeCursor = headNode.getNext();
+            IntNode nodeCursor = headNode;
             int i = 0;
-            // Go to the node at the nElements + 1 place
-            while (i < nElements - 1 && nodeCursor.getNext() != null) {
+            // Go to the node at the nElements place
+            while (i < nElements - 1 && nodeCursor.getNext() != null && nElements > 1) {
                 nodeCursor = nodeCursor.getNext();
                 i++;
             }
             // Set the nElements + 1 node as the root node
-            setHeadNode(nodeCursor);
+            // It might be null
+            setHeadNode(nodeCursor.getNext());
         }
     }