알고리즘/프로그래머스

[탐욕법(Greedy)] 체육복

래울 2021. 10. 9. 18:09

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

 

코딩테스트 연습 - 체육복

점심시간에 도둑이 들어, 일부 학생이 체육복을 도난당했습니다. 다행히 여벌 체육복이 있는 학생이 이들에게 체육복을 빌려주려 합니다. 학생들의 번호는 체격 순으로 매겨져 있어, 바로 앞번

programmers.co.kr

 

문제를 꼼꼼히 안 읽어서 고생했다.

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