From c2799fec3afd6010e36d8f56d6277701782eba41 Mon Sep 17 00:00:00 2001
From: marina <138340846+bt3gl-cryptography@users.noreply.github.com>
Date: Mon, 31 Jul 2023 10:54:14 -0700
Subject: [PATCH] Update random_set.py

---
 sets/random_set.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/sets/random_set.py b/sets/random_set.py
index 3dc2a4c..2a11f6e 100644
--- a/sets/random_set.py
+++ b/sets/random_set.py
@@ -19,8 +19,8 @@ class RandomizedSet:
 
     def remove(self, val: int) -> bool:
         if val in self.dict:
-            index = self.dict[val]
-            self.set[index] = self.set[-1]
+            last_element, idx = self.set[-1], self.dict[val]
+            self.set[idx], self.dict[last_element] = last_element, idx
             self.set.pop()
             del self.dict[val]
             return True