#프로그래머스 파이썬 Lv.1 #체육복

문제설명
: 전체 학생의 수 n, 체육복을 도난당한 학생들의 번호가 담긴 배열 lost, 여벌의 체육복을 가져온 학생들의 번호가 담긴 배열 reserve가 매개변수로 주어질 때, 체육수업을 들을 수 있는 학생의 최댓값을 return 하도록 solution 함수를 작성

문제풀이

1. set을 통해 reserve 와 lost 의 차집합을 구한다
2. reserve 에서 lost에 겹치지 않는 것을 비교하는데
3. 바로 앞 번호의 학생이나 바로 뒷 번호의 학생에게만 체육복을 빌려줄 수 있기 때문에
4. 앞번호나 뒷번호에 체육복을 잃어벼렀으면 빌려준 학생수를 올리고  잃어버린 항목에서 해당 학생을 지워준다
5. 모든 학생이 체육복을 빌릴 수 있으면 전체 학생수 n을 출력하고
6. 아니면 빌릴 수 있는 학생 수의 최대 학생 수를 출력.

def solution(n, lost, reserve):
    set_reserve = set(reserve) - set(lost)
    set_lost = set(lost) - set(reserve)

    for i in set_reserve :
        if i-1 in set_lost :
            set_lost.remove(i-1)
        elif i+1 in set_lost :
            set_lost.remove(i+1)
    return n - len(set_lost)

 

# 다른사람 풀이

1. 처음 _reserve와 _lost배열을 만들면서 중복을 제거
2. 제한사항을 not in 으로 처리
3. 여벌옷을 가진 학생들의 배열 원소를 하나씩 뽑아서 왼쪽이나 오른쪽으로 인덱스를 이동했을 때, 옷을 0개 가진 배열에 해당 인덱스가 존재하면 빌려주는 식으로 for 문을 돌며 forward, back 후 remove
4.  결국 _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)

 

 

 

 

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

코딩테스트 연습 - 체육복