[프로그래머스] Greedy 문제 체육복

문제풀러가기

프로그래머스

내 풀이

def solution(n, lost, reserve):
    answer = n
    loster = []
    for l in lost:
        if l in reserve:
            reserve.remove(l)
        else:
            loster.append(l)

    for l in loster:
        if (l-1) in reserve:
            reserve.remove(l-1)
        elif (l+1) in reserve:
            reserve.remove(l+1)
        else:
            answer = answer - 1

    return answer

탐욕법에 관련된 문제입니다. 문제를 잘못 읽어서 시간낭비를 많이 했네요. ㅠㅠ
도둑 맞은 사람이 여유복이 있을 경우 그냥 잃어버린 사람도, 여유복이 있는 사람도 아니랍니다. 혹시 앞 뒤 다른 사람것을 빌린 뒤
여유옷이 생겼다라고 착각하시면 문제가 안풀립니다. 따라서 처음에 lost와 reserve를 다시 만든 후 생성해야합니다.
list.remove()를 사용했는데 for문에 있는 리스트를 삭제해버리면 for문이 제대로 반복이 안되니 remove사용 시 신중하시길 바랍니다.

---

참고 풀이 

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]

다른 분 풀이 중 선언과 정의를 한번에 한 풀이 상단 부분인데 깔끔하네요