Add code checking same students

grootwo · web-flow · commit b90ab99d8b29 · 2023-07-22T22:53:12.000+09:00
diff --git a/programmers/Lv.1/max_training_cloth.py b/programmers/Lv.1/max_training_cloth.py
@@ -1,7 +1,10 @@
 def solution(n, lost, reserve):
     count = 0
     for num in lost:
-        if num - 1 in reserve:
+        if num in reserve:
+            reserve.remove(num)
+            count += 1
+        elif num - 1 in reserve:
             reserve.remove(num - 1)
             count += 1
         elif num + 1 in reserve:
