[프로그래머스] Level1 체육복(with python)

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

프로그래머스

 

1. 내가 작성한 코드

def solution(n, lost, reserve):
    
    new_lost = set(lost) - set(reserve)
    new_reserve = set(reserve) - set(lost)
    
    for i in new_lost:
        if i-1 in new_reserve:
            new_reserve.remove(i-1)
        elif i+1 in new_reserve:
            new_reserve.remove(i+1)
        else:
            n -= 1
            
    return n

 

문제를 읽지 않고 풀어서 해결하는 데 시간이 좀 많이 걸렸다.

 

혹시 일부 테스트케이스가 틀려서 검색중이시라면, 여벌의 체육복을 가지고 온 학생도 체육복을 잃어버릴 수 있다는 조건을 가지고 다시 풀어보시길 추천

 

set함수를 이용해서 여벌의 체육복이 있지만 잃어버린 학생을 먼저 제거해주고, 여벌의 체육복도 없고 체육복도 잃어버린 학생들을 처리해주면 해결된다.

 

문제 꼼꼼히 읽어야지,,,ㅠㅠ