[프로그래머스/Level1/파이썬3(python3)] 체육복

[프로그래머스/Level1/파이썬3(python3)] 체육복

문제

점심시간에 도둑이 들어, 일부 학생이 체육복을 도난당했습니다. 다행히 여벌 체육복이 있는 학생이 이들에게 체육복을 빌려주려 합니다. 학생들의 번호는 체격 순으로 매겨져 있어, 바로 앞번호의 학생이나 바로 뒷번호의 학생에게만 체육복을 빌려줄 수 있습니다. 예를 들어, 4번 학생은 3번 학생이나 5번 학생에게만 체육복을 빌려줄 수 있습니다. 체육복이 없으면 수업을 들을 수 없기 때문에 체육복을 적절히 빌려 최대한 많은 학생이 체육수업을 들어야 합니다.

 전체 학생의 수 n, 체육복을 도난당한 학생들의 번호가 담긴 배열 lost, 여벌의 체육복을 가져온 학생들의 번호가 담긴 배열 reserve가 매개변수로 주어질 때, 체육수업을 들을 수 있는 학생의 최댓값을 return 하도록 solution 함수를 작성해주세요.

제한사항

• 전체 학생의 수는 2명 이상 30명 이하입니다.
• 체육복을 도난당한 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
• 여벌의 체육복을 가져온 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
• 여벌 체육복이 있는 학생만 다른 학생에게 체육복을 빌려줄 수 있습니다.
• 여벌 체육복을 가져온 학생이 체육복을 도난당했을 수 있습니다. 이때 이 학생은 체육복을 하나만 도난당했다고 가정하며, 남은 체육복이 하나이기에 다른 학생에게는 체육복을 빌려줄 수 없습니다.

입출력 예

n	lost	reserve	return
5	[2, 4]	[1, 3, 5]	5
5	[2, 4]	[3]	4
3	[3]	[1]	2

입출력 예 설명

예제 #1 1번 학생이 2번 학생에게 체육복을 빌려주고, 3번 학생이나 5번 학생이 4번 학생에게 체육복을 빌려주면 학생 5명이 체육수업을 들을 수 있습니다.

예제 #2 3번 학생이 2번 학생이나 4번 학생에게 체육복을 빌려주면 학생 4명이 체육수업을 들을 수 있습니다.

풀이

def solution(n, lost, reserve):
    set_reserve = set(reserve) - set(lost)
    set_lost = set(lost) - set(reserve)
    for i in set_reserve:
        if i-1 in set_lost:
            set_lost.remove(i-1)
        elif i+1 in set_lost:
            set_lost.remove(i+1)
    return n - len(set_lost)

설명

내 방식대로 풀었을 땐 제출을 하면 꼭 테스트 12에서만 오답처리가 되던데 뭐가 문제인지 모르겠어가지고 RAINBOW-LAB에서 참고하였다. 아니 사실상 참고가 아니라 카피지... 아무튼 이 문제에서 중요한 건 이 문제가 탐욕법으로 분류돼있다는 것이고, 탐욕법 문제가 항상 그러하듯이 정공법으로 풀어내야 한다는 점이다.

 그 다음으로 주목해야할 점은 잔여체육복을 가져온 reserve로 입력된 학생이더라도 한 개를 도난당해서 수업은 들을 수 있지만 빌려줄 순 없는 인원이 있고, lost로 입력된 학생이더라도 여분을 갖고 왔기 때문에 수업을 들을 수 있는 학생이 있다는 점이다. 무슨 말장난인가, 결국 같은 얘기 아닌가 싶지만 알고리즘을 풀 땐 다른 용도로 사용하기 때문에 각각 필요한 요소로 구분해야 한다.

 일단 두번째 제한사항에서 "중복되는 번호는 없다"는 조건이 있으므로 set을 쓸 수 있다. 그렇게 set_reserve = set(reserve) - set(lost)를 통해 여분을 가져온 학생들 중 체육복을 빌려줄 수 있는 학생들의 번호만 남기고 set_lost = set(lost) - set(reserve)를 통해 체육복을 잃어버린 학생들 중 여분이 없어 진짜 체육복을 빌려야 하는 학생들을 구한다. 그리고 여분이 남은 학생들의 번호 앞뒤로 조회하여 체육복이 진짜 없는 학생들이 있으면 빌려주고 진짜 잃어버린 set에서 제외시킨다. 그렇게 총 학생수에서 "진짜 체육복이 없는 학생들의 수"를 빼면 정답이 되는 것이다.

p.s: 꼭 제출을 누르면 테스트케이스 한 두개만 박살나는 경우가 아주 가끔씩 있는데 그럴 때마다 입력되는 값들이 어떤 건지 진짜 너무 궁금하다. 하긴 근데 그걸 노출해버리면 조건문으로 모든 입력값에 대응하는 return값을 일일히 입력하여 중국어 방 형식으로 풀어버릴 수 있기 때문에 이해는 간다. 정진해야겠다.