Written by
최태열
on
on
[Python][프로그래머스][Level1] 체육복
[Python][프로그래머스][Level1] 체육복
코드
def solution(n, lost, reserve):
lost2 = [i for i in lost if not i in reserve]
reserve2 = [i for i in reserve if not i in lost]
answer = n - len(lost2)
for i in lost2:
if i-1 in reserve2:
reserve2.remove(i-1)
answer+=1
elif i+1 in reserve2:
reserve2.remove(i+1)
answer+=1
return answer
여분이 없으면서 잊어버린 index만 lost에 남긴다.
체육복이 2개 있는 index만 reserve에 남긴다.
양 옆 중에 reserve에 있는 값이 있다면 answer에 1을 더한다.
Discussion and feedback