Progammers 프로그래머스 - [체육복] Python 파이썬 풀이

 

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

📝 문제

 

🚫 제한 사항

 

✏️ 입출력

 

💡 풀이

 

문제 조건에서 '여벌 체육복을 가져온 학생이 도난당했을 수 있다' 라고 했기 때문에 lost와 reserve에서 중복되는 것들은 제거해야 한다. 

처음에는 그냥 for문에서 중복 여부만 따져 remove를 사용해 제거했다. 제출하니 90%만 맞고 일부 테스트케이스에는 실패가 나왔다. 

해결 방법은 배열을 새로 생성해 삭제하지 않을 번호들만 넣어 배열을 다시 갱신하는 것이다. 

나는 new_lost를 새로 생성해 reserve에 없는 번호들은 새 배열에 저장했다. 

그리고 자신을 기준으로 i-1, i+1을 확인해 체육복을 빌릴 수 있다면 lost에서 해당 번호를 지웠다. 

 

def solution(n, lost, reserve):
    answer = 0
    lost.sort()
    reserve.sort()
    
    # 배열을 새로 생성해 삭제하지 않을 번호들만 넣어 배열을 다시 갱신
    new_lost = []
    for x in lost:
        if x in reserve:    
            reserve.remove(x)
        else:
            new_lost.append(x)
    lost = new_lost

	
    # 자신을 기준으로 왼쪽, 오른쪽 친구에게 빌릴 수 있는지 확인
    for i in reserve :
        if (i-1) in lost : 
            lost.remove(i-1)

        elif (i+1) in lost :
            lost.remove(i+1)

    answer = n - len(lost)
    return answer