[일일코테 Day6] 체육복 - 프로그래머스(레벨1)

문제

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

코딩테스트 연습 - 체육복

풀이

 

코딩

function solution(n, lost, reserve) {
    const students = [];
    for (let i=0; i<n; ++i) {
        students.push(1);
    }
    
    reserve.forEach(r => {students[r-1] = 2})
    lost.forEach(l => {students[l-1] -= 1})
    
    
    let answer = 0;
    students.forEach((s, i, students) => {
        if (s > 0) {
            answer++;
            return;
        }
        
        if (i > 0 && students[i-1] === 2) {
            students[i-1] = 1;  // borrow from before
            students[i] = 1;
            answer += 1;
        } else if (i < students.length-1 && students[i+1] === 2) {
            students[i+1] = 1;  // borrow from next
            students[i] = 1;
            answer += 1;
        }
    });
    return answer;
}

실수

- 여분 체육복을 가지고 온 학생도 도난당할 수 있다는 제약 사항을 잘 확인 하지 못했다.