알고리즘/프로그래머스
[파이썬][프로그래머스] 탐욕법(Greedy) 체육복
SBOX Learning by doing
2022. 12. 2. 21:45
반응형
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
반응형