[Python] [프로그래머스] 체육복

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

코딩테스트 연습 - 체육복

 

def solution(n, lost, reserve):
    
    student = {}
    for i in range(n):
        student[i + 1] = 1
    for l in lost:
        if l in  student:
            student[l] -= 1
    for r in reserve:
        if r in  student:
            student[r] += 1

    answer = 0
    
    for key in student.keys():
        if student[key] == 0 and key == 1:
            if student[key + 1] == 2:
                student[key] += 1
                student[key + 1] -= 1
        elif student[key] == 0 and (2 <= key <= n - 1):
            if student[key - 1] == 2:
                student[key - 1] -= 1
                student[key] += 1
            elif student[key + 1] == 2:
                student[key + 1] -= 1
                student[key] += 1
        elif student[key] == 0 and key == n:
            if student[key - 1] == 2:
                student[key - 1] -= 1
                student[key] += 1
            
    for i in student.values():
        if i >= 1:
            answer += 1
    return answer

 

 

def solution(n, lost, reserve):
     answer = 0
     for i in range(1, n + 1):
         if i not in lost:
             answer += 1
         elif i in reserve:
             answer += 1
             lost.remove(i)
             reserve.remove(i)
            
     for i in lost:
         if i - 1 in reserve:
             answer += 1
             reserve.remove(i - 1)
         elif i + 1 in reserve:
             answer += 1
             reserve.remove(i + 1)

    return answer