[프로그래머스] 체육복

0. 제목

• 프로그래머스 체육복

---

1. 문제

https://programmers.co.kr/learn/courses/30/lessons/42862

코딩테스트 연습 - 체육복

---

2. 풀이

• 잃어버린 사람과 여분있는 사람이 같은경우
• 같지 않은경우는 -1, +1이 있는경우로 경우를 나눈다.
• 총 인원수 - lost수에서 여분수를 더해서 최종값을 구한다.

 

---

3. 코드

def solution(n, lost, reserve):
    answer = n-len(lost)
    before=len(reserve)
    lost_copy=[]

    for i in lost:
        lost_copy.append(i)

    for i in lost_copy:
        if i in reserve:
            reserve.remove(i)
            lost.remove(i)

    for i in lost:
        if i-1 in reserve:
            reserve.remove(i-1)
        elif i+1 in reserve:
            reserve.remove(i+1)


    after=len(reserve)
    answer+=before-after

    return answer