programmers | Lv1. 체육복 [Python]

💡 Python 3

🔗 문제

체육복 [Link]

💻 코드

def solution(n, l, r):
    answer = 0
    for i in range(1, n+1):
        if i in l:
            if i in r: 
                answer += 1
                r.remove(i)
            elif i-1 in r and i-1 not in l:
                answer += 1
                r.remove(i-1)
            elif i+1 in r and i+1 not in l:
                answer += 1
                r.remove(i+1)
        else:
            answer += 1    
    return answer

💥 다른 사람 코드

def solution(n, lost, reserve):
    _reserve = [r for r in reserve if r not in lost]
    _lost = [l for l in lost if l not in reserve]
    for r in _reserve:
        f = r - 1
        b = r + 1
        if f in _lost:
            _lost.remove(f)
        elif b in _lost:
            _lost.remove(b)
    return n - len(_lost)