[프로그래머스]체육복

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

프로그래머스

---

문제 조건 중, 이 마지막 조건만 주의하면 될 것 같다.

 

여벌 체육복을 가져온 학생이 체육복을 도난당했을 경우, 다른 학생에게 빌려줄 수 없다.

 --> 해당 학생들을 먼저 처리한 다음, 다른 도난당한 학생들에게 체육복을 빌려주는 작업을 진행해야 정답 판정을 받을 수 있었다.

---

python

def solution(n, lost, reserve):
    students = [0]*(n+1) # 학생 번호는 1부터 시작
    
    # 잃어버린 학생이 여벌체육복 가지고 있는 경우
    for i in range(1,n+1):
        if i in lost and i in reserve:
            lost.remove(i)
            reserve.remove(i)
            
    for i in range(1,n+1):
        if i not in lost:
            students[i] = 1
        else:
            if i-1 in reserve: # 이전 학생
                reserve.remove(i-1)
                students[i] = 1
            elif i+1 in reserve: # 다음 번호 학생
                reserve.remove(i+1)
                students[i] = 1
    answer = students.count(1)
    return answer