본문 바로가기
알고리즘/프로그래머스

[파이썬][프로그래머스] 탐욕법(Greedy) 체육복

by SBOX Learning by doing 2022. 12. 2.
반응형

def solution(n, lost, reserve):
    haveTrainning = []
    lost.sort()
    reserve.sort()
    # 여유분이 있지만 잃어버린 학생
    temp = []
    for i in lost:
        if i in reserve:
            temp.append(i)
    
    for i in temp:
        lost.pop(lost.index(i))
        reserve.pop(reserve.index(i))

    for i in range(1, n+1):
        if i not in lost:
            haveTrainning.append(i)

    for i in lost:
        if (i-1) in reserve:
            reserve.pop(reserve.index(i-1))
            haveTrainning.append(i)
        else:
            if (i+1) in reserve:
                reserve.pop(reserve.index(i+1))
                haveTrainning.append(i)
                
    answer = len(haveTrainning)
    return answer

반응형

댓글