01.py

# 문제 설명
# 점심시간에 도둑이 들어, 일부 학생이 체육복을 도난당했습니다. 다행히 여벌 체육복이 있는 학생이 이들에게 체육복을 빌려주려 합니다. 학생들의 번호는 체격 순으로 매겨져 있어, 바로 앞번호의 학생이나 바로 뒷번호의 학생에게만 체육복을 빌려줄 수 있습니다. 예를 들어, 4번 학생은 3번 학생이나 5번 학생에게만 체육복을 빌려줄 수 있습니다. 체육복이 없으면 수업을 들을 수 없기 때문에 체육복을 적절히 빌려 최대한 많은 학생이 체육수업을 들어야 합니다.
# 전체 학생의 수 n, 체육복을 도난당한 학생들의 번호가 담긴 배열 lost, 여벌의 체육복을 가져온 학생들의 번호가 담긴 배열 reserve가 매개변수로 주어질 때, 체육수업을 들을 수 있는 학생의 최댓값을 return 하도록 solution 함수를 작성해주세요.

# 제한사항
# 전체 학생의 수는 2명 이상 30명 이하입니다.
# 체육복을 도난당한 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
# 여벌의 체육복을 가져온 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
# 여벌 체육복이 있는 학생만 다른 학생에게 체육복을 빌려줄 수 있습니다.
# 여벌 체육복을 가져온 학생이 체육복을 도난당했을 수 있습니다. 이때 이 학생은 체육복을 하나만 도난당했다고 가정하며, 남은 체육복이 하나이기에 다른 학생에게는 체육복을 빌려줄 수 없습니다.
# 입출력 예
# n	    lost	    reserve	    return
# 5	    [2, 4]	    [1, 3, 5]	5
# 5	    [2, 4]	    [3]	        4
# 3	    [3]	        [1]	        2

# 입출력 예 설명
# 예제 #1 : 1번 학생이 2번 학생에게 체육복을 빌려주고, 3번 학생이나 5번 학생이 4번 학생에게 체육복을 빌려주면 학생 5명이 체육수업을 들을 수 있습니다.
# 예제 #2 : 3번 학생이 2번 학생이나 4번 학생에게 체육복을 빌려주면 학생 4명이 체육수업을 들을 수 있습니다.

# 나의 정답2
def solution(n, lost, reserve):
    has_reserve = [False]*(n+2)
    for r in reserve:
        has_reserve[r] = True
    
    lost.sort()
    
    for l in list(lost):
        if has_reserve[l]:
            lost.remove(l)
            has_reserve[l] = False
    
    has_cloth = n - len(lost)
    
    for l in lost:
        if has_reserve[l-1]: 
            has_cloth += 1
            continue
        if has_reserve[l+1]: 
            has_cloth += 1
            has_reserve[l+1] = False
            continue
            
    return has_cloth

# =================================================================
# 나의 정답1
def solution(n, lost, reserve):
    lost.sort()
    
    for l in lost.copy():
        if l in reserve:
            reserve.remove(l)
            lost.remove(l)

    for l in lost.copy(): # 원본 lost 리스트에서 도난당했지만 본인 여벌이 있는 경우 제외된 값
        if l-1 in reserve:            
            reserve.remove(l-1)
        elif l+1 in reserve:            
            reserve.remove(l+1)
        else:
            continue
        lost.remove(l)
            
    return n - len(lost)

# =================================================================
# 다른 사람의 풀이 
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]
    for r in _reserve:
        f = r - 1
        b = r + 1
        if f in _lost:
            _lost.remove(f)
        elif b in _lost:
            _lost.remove(b)
    return n - len(_lost)
    
# =================================================================