[프로그래머스]체육복

문제 링크 : https://school.programmers.co.kr/learn/courses/30/lessons/42862

프로그래머스

문제를 간단히 요약하면 다음과 같다.

체육복이 없는 인원이 담긴 lost 배열과 여별의 체육복을 가진 인원이 담긴 reverse 배열, 인원 수가 담긴 정수 n이 주어진다.

여벌의 체육복이 있는 인원이 체육복이 없는 인원에게 체육복을 빌려줄 수 있다.

하지만 각 번호에서 +-1을 한 숫자에게만 체육복을 빌려줄 수 있을 때, 최대 몇 명이 체육복을 입을 수 있는지 반환해야한다.

 

코드를 작성하기 위한 순서는 다음과 같다.

1. 문제에서 여벌 옷이 있는 학생도 잃어버린 명단에 있을 수 있다라고 명시되어 있기 때문에 중복된 값을 제거해준다.

2. 잃어버린 사이즈에서 +1을 먼저 비교하는 것보다 -1을 먼저 비교해야 최대한 많은 인원에게 체육복을 빌려줄 수 있기 때문에, -1을 한 값이 있는지 비교한 후, +1을 한 값이 있는지 비교한다.

3. 둘 다 해당하는 값이 없다면 체육복을 빌려줄 수 없기에 전체 인원에서 -1을 해준다.

def solution(n, lost, reserve):
    new_lost = set(lost) - set(reserve) #1번
    new_reserve = set(reserve) - set(lost) #1번
    for i in new_lost:
        if i - 1 in new_reserve: #2번
            new_reserve.remove(i - 1)
        elif i + 1 in new_reserve:
            new_reserve.remove(i + 1)
        else: #3번 : 체육복을 빌려줄 수 없는 상황
            n-=1
    return n

해당 문제는 '여벌 옷이 있는 학생도 잃어버린 명단에 있을 수 있다.'라는 내용을 꼭 알아야 풀 수 있는 문제였다.