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

코딩테스트 연습 - 체육복 | 프로그래머스 (programmers.co.kr)

코딩테스트 연습 - 체육복

function solution(n, lost, reserve) {
    let answer = 0;

    let currentArray = Array.from({length:n}, ()=>1);
    
    //set
    inputCount(reserve, currentArray, 1);
    inputCount(lost, currentArray, -1);
    //logic
    for(let i=0; i<currentArray.length; i++){
        if(currentArray[i]==2 && currentArray[i-1] == 0) {
            currentArray[i-1]++;
            currentArray[i]--;
            continue;
        }
        if(currentArray[i]==2 && currentArray[i+1] == 0){
            currentArray[i+1]++;
            currentArray[i]--;
        }
    }
    for(let i=0; i<currentArray.length; i++){
        if(currentArray[i] > 0) answer++;
    }
    //set function
    function inputCount(keys, array, value){
        for(let i=0; i<array.length; i++){
            for(let j=0; j<keys.length; j++){
                if((keys[j]-1) == i) array[i] +=value;
            }
        }
        
    }
    return answer;
}