[프로그래머스] 자바스크립트 '체육복' 문제풀이 (LV.1)

[프로그래머스] 자바스크립트 '체육복' 문제풀이 (LV.1)

문제

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/42862?language=javascript

프로그래머스

풀이 알고리즘: 그리디

난이도: LV.1

풀이 코드 💡

function solution(n, lost, reserve) {
    let answer = n - lost.length;
    const realLost = lost.sort((a,b) => a  - b).filter(item => !reserve.includes(item));
    answer += lost.length - realLost.length;
    const realReserve = reserve.sort((a,b) => a-b).filter(item => !lost.includes(item));

    for(let i = 0; i < realLost.length; i++) {
        for(let j = 0; j < realReserve.length; j++){
            if(Math.abs(realLost[i] - realReserve[j]) === 1) {
                answer++;
                realReserve.splice(j, 1);
                break;
            } 
        }
    }

    return answer;
}

반응형

풀이 해설 💡

해당 문제는 그리디로 구현할 수 있는 문제다.

먼저 lost와 reserve배열을 오름차순 정렬을 해준 후에 lost와 reserve 배열에서 겹치는 값이 없는 값들로만 새로 만들어 주었다.

 

그 이유는 다양한 테스트 케이스가 발생하기 떄문인데,

 

오름차순 정렬을 안 했을 시에는 lost와 reserve 배열의 값들이 오름차순으로 입력이 주어질 거란 보장이 없기 떄문에 예외가 발생한다.

 

겹치는 값들을 처리 안 해주게 되면

n	lost	reserve	Return
5	[4, 5]	[3, 4]	4

위와 같은 입력이 주어졌을 때 통과를 하지 못하게 된다.

 

그리고 새로 추출한 lost 배열을 기준으로 반복문을 돌린다.

해당 반복문 안에 reserve 배열도 같이 반복문을 실행하게 되는데 문제에서 자신의 앞 또는 뒤에 학생에게만 빌려줄 수 있기 때문에 잃어버린 사람과 빌려주려는 사람의 차이가 1만큼 나면 빌려줄 수 있게 되는 것이다.

둘의 차이가 1만큼 나면 answer를 카운트해주고 reserve에서 빌려준 값을 삭제한다.

값을 삭제하는 이유는 reserve 반복문은 0부터 시작하게 되는데 삭제를 해주지 않으면 빌려준 사람이 또 빌려주게 되는 상황이 발생하게 된다.

잃어버린 학생에 대해서 체육복을 빌려주게되면 reserve 반복문에서 break를 함으로써 빠져나온다.

결과 제출 ✅

코드 제출 결과

해당 문제는 문제에서 제시해 주는 테스트케이스가 직접 제출 했을 때 테스트케이스에 비해서 정보가 적어 예외 사항이 많이 발생하기 때문에 골머리를 앓은 문제였다.