[JS] 프로그래머스 : 체육복

 

Question

 

 

입출력 예시

 

입출력 예시 설명

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

 

예제 #2
3번 학생이 2번 학생이나 4번 학생에게 체육복을 빌려주면 학생 4명이 체육수업을 들을 수 있습니다.

 

---

 

 

My Code

 

참고 : 코드의 진화과정을 볼 수 있다  😎

 

 

 

 

 

코드 실행했을 때 통과했지만 테스트 케이스에 걸려 실패한 코드이다.

function solution(n, lost, reserve) {
    let lostLength = lost.length;
    let answer =  n - lostLength;
    let arr = [];

    for (let i = 0 ; i < lostLength ; i ++) {
        if (reserve.includes(lost[i] - 1 )) {
            answer += 1 ;
            reserve = reserve.filter ((ele) => ele !== lost[i] - 1 )
        }
        else if (reserve.includes(lost[i] + 1)) {
            answer += 1;
            reserve = reserve.filter((ele) => ele !== lost[i] + 1)
        }
    }
    return answer;
}

 

WHY?

lost와 reserve에 둘다 속해있는 학생수를 고려하지 않았다.

만약에 둘다 속해 있다면 결국에는 빌려줄 체육복은 없고 자신의 체육복만 입을 수 있는 경우이므로

reserve와 lost에서 삭제하고 체육수업을 들을 수 있는 인원수에 추가해야 한다.

항상 이런 경우의수를 까먹는다! 잊지 말자!

 

 

 

추가했는데 또 틀렸다.

function solution(n, lost, reserve) {
    let lostLength = lost.length;
    let answer =  n - lostLength;
    
    // 체육복도 잃어버리고 여분의 체육복도 동시에 있는 경우
    for (let i = 0; i < lostLength ; i ++) {
        if (reserve.includes(lost[i])) {
            reserve = reserve.filter((ele) => ele !== lost[i])
            answer += 1;
        } else {
            tmp.push(lost[i]);
        }
    }

    for (let i = 0 ; i < lostLength ; i ++) {
        if (reserve.includes(lost[i] - 1 )) {
            answer += 1 ;
            reserve = reserve.filter ((ele) => ele !== lost[i] - 1 )
        }
        else if (reserve.includes(lost[i] + 1)) {
            answer += 1;
            reserve = reserve.filter((ele) => ele !== lost[i] + 1)
        }
    }
    return answer;
}

 

WHY?

구글링을 해보니 테스트 케이스가 정렬해서 들어오지 않는 경우가 있다고 한다.

아니 테스트 케이스야 그런 거까지 일일이 말 안 해주었잖니?

내가 그걸 어떻게 아니? 🤷‍♂️

sort() 함수를 추가하고 모든 테스트 케이스를 통과했다.

 

 

 

아래가 모든 테스트케이스를 통과한 코드!

function solution(n, lost, reserve) {
    let lostLength = lost.length;
    let answer =  n - lostLength;
    let tmp = [];
    
    // 체육복도 잃어버리고 여분의 체육복도 동시에 있는 경우
    for (let i = 0; i < lostLength ; i ++) {
        if (reserve.includes(lost[i])) {
            reserve = reserve.filter((ele) => ele !== lost[i])
            answer += 1;
        } else {
            tmp.push(lost[i]);
        }
    }
    // 순서에 상관없이 들어올 입력을 대비해 정렬해줌
    lost = tmp.sort();
    
    for (let i = 0 ; i < lostLength ; i ++) {
        if (reserve.includes(lost[i] - 1 )) {
            answer += 1 ;
            reserve = reserve.filter ((ele) => ele !== lost[i] - 1 )
        }
        else if (reserve.includes(lost[i] + 1)) {
            answer += 1;
            reserve = reserve.filter((ele) => ele !== lost[i] + 1)
        }
    }
    return answer;
}

 

 

 

 

 

다른 사람의 풀이 중에 엄청난 걸 발견했다!

가독성도 좋고 진짜 깔끔한 코드!

초보인 내가 봐도 괜찮은 코드!

그래서 가져와봤다.

(혹시 문제가 된다면 댓글로 꼭 알려주시길)

 

function solution(n, lost, reserve) {
    const realReserve = reserve.filter(r => !lost.includes(r));
    const realLost = lost.filter(r => !reserve.includes(r));

    // const reserveNum = reserve.lenght - realReserve.length;

    const ableNum = realLost.filter(a => {
        return realReserve.find((b, i) => {
            const has = b === a-1 || b === a+1;
            if (has) {
                delete realReserve[i];
            }
            return has;
        });
    }).length;
    return n - (realLost.length - ableNum);
}

출처 : 프로그래머스 / - , chifuyu1 , doi-h , Five1star , seo2im 외 6명이 푼 코드

 

 

 

이 코드를 왜 좋다고 느낀 것일까?

 

• 가독성이 좋다.
• 메서드를 적절하게 활용했다.
• 중복되거나 비슷한 코드가 없다.
• 어떤 기능을 수행하는지 이해가 간다.
• delete라는 연산자를 알아냈다.

 

 

 

추가된 테스트 케이스들 중에 통과 안 되는 부분이 있지만

코드를 이렇게 쓸 수도 있구나 배우게 됐다.

나도 활용할 수 있도록 계속해서 많은 코드를 봐야겠다.

코드에 정답이 없다는 게 재밌다.

내가 만들어 낼 수 있다는 것을 알고 간다.