[프로그래머스][파이썬/Python] 체육복

문제 설명

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

 

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

 

제한사항

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

 

입출력 예

 

풀이 과정

1. 최종 제출 코드와 테스트 결과 (통과)

def solution(n, lost, reserve):
    # case 1 : No lost
    answer = n - len(lost)

    # case 2 : lost
    for i in lost:
        if i in reserve:
            answer = answer + 1
            reserve.remove(i)
            continue
        for j in reserve:
            if j == i-1:
                answer = answer + 1
                reserve.remove(j)
                break
            elif j == i+1:
                if j in lost:
                    break
                answer = answer + 1
                reserve.remove(j)
                break

    return answer

문제를 풀면서 가장 힘들었던 부분은 테스트 케이스가 뭔지 모른다는 것이었다. 대체 뭐길래 자꾸 실패를 하는지 알고 싶었다. 하지만 현실이었다면 내가 스스로 알아내야 하는 것이기 때문에, 테스트 케이스를 생각해내는 것은 나의 능력이라고 생각하며 불평을 멈췄다.

 

프로그래머스 코딩테스트 화면 좌측 하단에 '질문하기' 버튼이 있는데, 눌러서 들어가보니 다들 나와 같이 12번 케이스로 인해 많이 힘들어했다. 그런데 거기서 김지훈님의 "12번 테스트 케이스 실패 시 케이스 추가해보세요"라는 제목의 글과, JeongminYu님의 "Test 12번 에러"라는 제목의 글의 내용들을 확인하니 세상에 마상에. 이런 케이스가 있었구나 싶었다. 결국 남의 도움을 얻고 푼 꼴이기 때문에 나의 능력 부족이고, 사실상 내가 푼 것이 아니라고 생각한다. 테스트 케이스를 알고 나니 코드 구현하는 것은 너무나도 간단했지만 그럼 뭐하나. 케이스를 생각해내는 것이 이 문제의 핵심인 것 같다.

 

 

 

2. 통과 전 시도했던 코드들과 반성의 시간들

 

# 1차 코드

def solution(n, lost, reserve):
    # case 1 : No lost
    answer = n - len(lost)

    # case 2 : lost
    for i in lost:
        for j in reserve:
            if j == i-1:
                answer = answer + 1
                reserve.remove(j)
            elif j == i:
                answer = answer + 1
                reserve.remove(j)
            elif j == i+1:
                answer = answer + 1
                reserve.remove(j)

    return answer

테스트 케이스가 왜 실패했는지는 감을 잡지 못했으나, 이중 for문 안에서 answer에 1을 더해주게 될 경우 break를 걸어줘야 reserve의 다른 원소를 추가적으로 살피지 않을 거란 생각을 했다. 그래서 위 코드에서 answer에 1을 더해주게 될 때마다 무의미한 탐색을 줄이고자 break를 걸어줬다.

 

 

# 2차 코드

def solution(n, lost, reserve):
    # case 1 : No lost
    answer = n - len(lost)

    # case 2 : lost
    for i in lost:
        for j in reserve:
            if j == i-1:
                answer = answer + 1
                reserve.remove(j)
                break
            elif j == i:
                answer = answer + 1
                reserve.remove(j)
                break
            elif j == i+1:
                answer = answer + 1
                reserve.remove(j)
                break

    return answer

그 결과, 12번 테스트 케이스를 제외하고는 모두 통과했다. 솔직히 테스트 12가 대체 뭐길래 틀리는거지 싶었고, 전혀 감이 오지 않았다. 고민하다가 문제를 다시 봤더니 제한 사항 중 간과하고 있던 게 있었다.

 

• 여벌 체육복을 가져온 학생이 체육복을 도난당했을 수 있습니다. 이때 이 학생은 체육복을 하나만 도난당했다고 가정하며, 남은 체육복이 하나이기에 다른 학생에게는 체육복을 빌려줄 수 없습니다.

위 조건에 의해 lost의 원소와 reserve의 원소가 같을 경우에는 다른 학생에게 체육복을 빌려줄 수 없기 때문에 가장 먼저 확인해야 한다. 제한 사항을 확인하는 것은 기본 중에 기본인데 정말 바보다. 아무튼, 위 조건을 고려하여 3차 코드를 작성했다.

 

 

# 3차 코드

def solution(n, lost, reserve):
    # case 1 : No lost
    answer = n - len(lost)

    # case 2 : lost
    for i in lost:
        if i in reserve:
            answer = answer + 1
            reserve.remove(i)
            continue
        for j in reserve:
            if j == i-1:
                answer = answer + 1
                reserve.remove(j)
                break
            elif j == i+1:
                answer = answer + 1
                reserve.remove(j)
                break

    return answer

이젠 통과하겠지 싶었는데 또 테스트 12번에서 걸렸다. 혼자 태블릿에 그림을 그려가며 '대체 무슨 경우인거야;' 끙끙 앓았는데 답이 나오지 않았다. 답답한 마음에 '질문하기' 버튼을 클릭했고, 거기서 힌트를 얻을 수 있었다. 특히 JeongminYu님의 "Test 12번 에러"라는 제목의 글이 결정적인 힌트가 되어 해당 문제를 통과할 수 있었다.

 

 

반성을 정리해보면,

1. 문제와 제한 사항을 꼼꼼하게 읽자.
2. 테스트 케이스를 더욱 창의적으로 생각해보자.

 

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