fix 380

fishercoder1534 · fishercoder1534 · commit 8c56452c7f89 · 2021-11-05T15:26:02.000-07:00
diff --git a/src/main/java/com/fishercoder/solutions/_380.java b/src/main/java/com/fishercoder/solutions/_380.java
@@ -36,12 +36,14 @@ public boolean remove(int val) {
                 if (!map.containsKey(val)) {
                     return false;
                 } else {
-                    int lastElement = list.get(list.size() - 1);
-                    int index = map.get(val);
-                    list.set(index, lastElement);
-                    map.put(lastElement, index);
-                    list.remove(list.size() - 1);
+                    int removeIndex = map.get(val);
+                    if (removeIndex != list.size() - 1) {//if it's not the last element, then we need to swap it with the last element so that this operation is also O(1)
+                        int lastElement = list.get(list.size() - 1);
+                        list.set(removeIndex, lastElement);
+                        map.put(lastElement, removeIndex);
+                    }
                     map.remove(val);
+                    list.remove(list.size() - 1);
                     return true;
                 }
             }
