[프로그래머스] 체육복(#1)

- https://huree-can-do-it.notion.site/6c5c2d075c0848deb05293a700a860cc
Alpagorithm · Dec 30, 2021 · b93c35d · b93c35d
1 parent 9c74dc7
commit b93c35d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/Programmers/main.py b/Programmers/main.py
diff --git a/Programmers/체육복.py b/Programmers/체육복.py
@@ -0,0 +1,12 @@
+# 풀이 참고
+def solution(n, lost, reserve):
+    # 여벌교복 가져온 학생도 도난 당할 수 있어서 중복삭제
+    reserveSet = set(reserve) - set(lost)
+    lostSet = set(lost) - set(reserve)
+
+    for i in reserveSet:
+        if i - 1 in lostSet:
+            lostSet.remove(i - 1)
+        elif i + 1 in lostSet:
+            lostSet.remove(i + 1)
+    return n - len(lostSet)