Solve py set discard remove pop

DucChuyenSoftwareEngineer · Apr 12, 2020 · 98bd10c · 98bd10c
1 parent 130df39
commit 98bd10c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/python/py-set-discard-remove-pop.py b/python/py-set-discard-remove-pop.py
@@ -0,0 +1,12 @@
+num_elements = int(input())
+s = set(map(int, input().split()))
+num_operations = int(input())
+for _ in range(num_operations):
+    operation = input().split(" ")
+    if(operation[0] == "pop"):
+        s.pop()
+    else:
+        op, val = operation
+        s.discard(int(val))
+
+print(sum(s))