체육복 (#2)

Alpagorithm · Dec 28, 2021 · 4738a5d · 4738a5d
1 parent b08b045
commit 4738a5d
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/Algorithm/프로그래머스/체육복.py b/Algorithm/프로그래머스/체육복.py
@@ -0,0 +1,19 @@
+n = 6
+lost = [2, 4]
+reserve = [1, 3, 5]
+
+set_lost = set(lost) - set(reserve)
+set_reserve = set(reserve) - set(lost)
+
+for i in set_reserve:
+    if i - 1 in set_lost:
+        set_lost.remove(i-1)
+        continue
+
+    if i + 1 in set_lost:
+        set_lost.remove(i+1)
+        continue
+
+answer = n - len(set_lost)
+
+print(answer)