[프로그래머스] 체육복(level1, python)

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

프로그래머스

아이디어

본문에 아래 부분을 고려해야 한다!!!! 

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

그래서 처음에는 전체 리스트에 1을 넣어준뒤, 여벌이 있는 사람의 값을 2로 바꿔주고, 도난당한 사람은 -1을 해주었다.

그리고 검사를 진행하면 아래 반례에도 제대로 작동한다. 

 

n = 5

lost = [1, 2]

reverse = [1, 2]

-> 결과는 5가 나와야 한다.

전체 코드

def solution(n, lost, reserve):
    arr = [1] * (n + 1)
    
    for r in reserve: arr[r] = 2
    for l in lost: arr[l] -= 1
    
    for idx, a in enumerate(arr):
        if a == 0:
            if idx > 1 and arr[idx-1] == 2:
                arr[idx-1] -= 1
                arr[idx] += 1
            elif idx < n and arr[idx+1] == 2:
                arr[idx+1] -= 1
                arr[idx] += 1
    return n - arr[1:].count(0)