[JS] 그리디 - 체육복

function solution(n, lost, reserve) {
  let answer = 0;
  let students = new Array(n + 1).fill(1);
  for (let i = 1; i < n + 1; i++) {
    if (lost.includes(i)) {
      students[i]--;
    }
    if (reserve.includes(i)) {
      students[i]++;
    }
  }

  for (let i = 1; i < n + 1; i++) {
    if (students[i] === 0) {
      if (students[i - 1] === 2) {
        students[i - 1]--;
        students[i]++;
      } else if (students[i + 1] === 2) {
        students[i + 1]--;
        students[i]++;
      }
    }
  }

  answer = students.filter((x) => x >= 1).length - 1;

  return answer;
}

 

프로그래머스

코딩테스트 연습 - 체육복

 

1. 학생들이 모두 1벌씩 옷을 갖고 있기에 1로 채운 배열을 만든다.(이 때, 배열의 0번째 인덱스는 무시한다.)
2. lost 배열에 들어있는 숫자를 인덱스로 하는 값은 --, 반대로 reserve내의 숫자에 대해선 ++ 해준다.
3. 배열엔 학생들마다 갖고 있는 옷의 숫자가 담겼고 문제 조건에 따라 학생의 옷이 0 일 때, 앞뒤 학생이 2벌이라면 각각 +1과 -1 처리
4. 쪼오금 개념이 헷갈릴 수 있는게, 0을 기준으로 앞뒤로 2를 찾는 순서가 정말 최선의 값이 나오게 하는걸까 싶을 수 있다. 하지만, 옷을 2벌 갖고 있는 학생의 입장에서 앞이든 뒤든 0인 친구가 있다면 무조건 옷을 주기만 하면 되므로 최선의 값이 나오게 된다.