[프로그래머스] 체육복

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

코딩테스트 연습 - 체육복

 

풀이

놓치기 쉬운 핵심 2가지만 신경쓰니까 바로 풀렸다

1. 도둑 맞은 학생이 여벌 옷이 있는 경우 : 미리 lost, reserve에서 소거
2. input 값이 정렬되지 않은 경우 : 순서에 상관없이 짜거나, sort

 

Code 1.

function solution(n, lost, reserve) {
    var answer = 0;
    const tmpLost = lost.slice();
    
    for (let idx in tmpLost) {
        const student = tmpLost[idx];
        if (hasReserveJersey(student, reserve)) {
            lost.splice(lost.indexOf(student), 1);
        }
    }

    for (let i = 1; i <= n; i++) {
        if (isLost(i, lost)) {
            const hasJersey = hasReserveJersey(i - 1, reserve) ||
                              hasReserveJersey(i + 1, reserve);
            if (!hasJersey) {
                continue;
            }
        }  
        answer ++;
    }
    return answer;
}

function isLost(n, lost) {
    return lost.find(e => e === n);
}

function hasReserveJersey(n, reserve) {
    const idx = reserve.indexOf(n);
    if (idx === -1) {
        return false;
    }
    
    reserve.splice(idx, 1);
    return true;
}

 

Code 2.

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

    const lostCount = realLost.filter(e => {
        return !hasReserveJersey(e - 1, realReserve) &&
               !hasReserveJersey(e + 1, realReserve);
    }).length;

    return n - lostCount;
}

function hasReserveJersey(n, reserve) {
    const idx = reserve.indexOf(n);
    if (idx === -1) {
        return false;
    }
    
    reserve.splice(idx, 1);
    return true;
}