[프로그래머스] 체육복

아니 안 푼 level1의 문제가 있었다니...ㅎㅎ

def solution(n, lost, reserve):
    answer = n - len(lost)
    student = set([i for i in range(1, n+1)]) - set(lost)

    for i in lost:
        if i in reserve:
            reserve.remove(i)
            answer = answer + 1
            student.add(i)
    lost = set(lost) - student        
    for i in lost:
        if (i-1) in reserve:
            reserve.remove(i-1)
            answer = answer + 1
        elif (i+1) in reserve:
            reserve.remove(i+1)
            answer = answer + 1

    return answer

1. 가장 우선순위가, 체육복을 잃어버렸는데 여분이 있는 사람은 누구에게도 빌려주지 않고 자기가 입는다. 이걸 처리해야했다.

그래서 이 처리만을 위한 for문을 한 번 돌았다. 그러면서 그 학생은 체육복이 있는 학생 array에 넣어주고, 여분 list에서는 빼준다.

 

2. 그 다음에 잃어버린 사람의 앞사람이 체육복을 가지고 있으면 그 학생을 체육복이 있는 학생 array에 넣어준다. 앞사람이 없으면 뒷사람은 있는지 검사한다.

 

 

<다른 사람의 풀이에서 배울점>

for문을 돌지 않고 애초에 생성을 할 때 

reserve_ = [i for i in reserve if i not in lost]
lost_ = [i for i in reserve if i not in reserve]

로 다시 생성을 하면, 1번과 같은 경우는 이미 처리가 된 채로 문제를 시작할 수 있다. 

시간복잡도 면에선 그게 그것인 것 같다. 그러나 훨씬 간결해보임.