문제

링크텍스트

난관

• 지난 스터디에서 시간이 부족했기에 서둘러 풀다보니 문제를 잘못이해했었다.
• 몸집이 작은 아이로 부터는 무조건 빌릴 수 있다고 생각한것.
• 신기한 건 그렇게 풀어도 'Run Test'는 통과했다는 점 ^^
• 결국 힌트를 통해 각 학생별로 앞이나 뒤에서 빌리는 걸로 변경하여 간신히 풀었다.

결과

def solution(n, lost, reserve):
    n_l = len(lost)
    n_r = len(reserve)
    reserve = sorted(reserve) 
    lost = sorted(lost)
    s = [1]*(n+2)
    s[0] = 0
    s[-1] = 0
    for i in lost:
        s[i] -= 1
    for i in reserve:
        s[i] += 1
    
    for i in range(1,n+1):
        if s[i] == 0 and s[i-1] == 2:
            s[i] += 1
            s[i-1] -= 1
        elif s[i] == 0 and s[i+1] == 2:
            s[i] += 1
            s[i+1] -= 1
    
    answer = 0
    for si in s:
        if si == 1 or si == 2:
            answer += 1
    return answer