[프로그래머스] 체육복 문제풀이 JS

lost와 reserve는 오름차순으로 정렬한다.

이 후 체육복을 잃어버렸는데 여분의 체육복이 있는 사람을 골라낸다.

이 경우엔 체육 수업을 들을 수 있는 경우이므로 count를 1증가하고,

해당 위치의 lost와 reserve는 -1로 표시한다.

 

lost와 reserve를 전체탐색해서

체육복을 잃어버린 사람의 앞번호 사람이 여분의 체육복이 있는지 먼저 체크하고,

아니라면 뒷번호 사람이 여분의 체육복이 있는지 체크한다.

여분의 체육복이 있어서 체육 수업을 들을 수 있는 경우가 되면 count를 1증가한다.

여분의 체육복을 준 경우라면 해당 위치의 reserve는 -1로 표시한다.

 

탐색 후, 전체 학생수에서 체육복을 잃어버린 학생의 수를 빼고,

여분의 체육복을 받아 체육 수업을 들을 수 있는 학생의 수 count를 더한 값을 반환하게 되면

원하는 결과를 얻게 된다. 

 

function solution(n, lost, reserve) {
  let count = 0
  lost.sort((a, b) => a - b)
  reserve.sort((a, b) => a - b)

  for (let i = 0; i < lost.length; i++) {
    for (let j = 0; j < reserve.length; j++) {
      if (lost[i] === reserve[j]) {
        lost[i] = -1
        reserve[j] = -1
        count++
      }
    }
  }
  for (let i = 0; i < lost.length; i++) {
    if (lost[i] !== -1) {
      for (let j = 0; j < reserve.length; j++) {
        if (lost[i] - 1 === reserve[j]) {
          count++
          reserve[j] = -1
          break
        } else if (lost[i] + 1 === reserve[j]) {
          count++
          reserve[j] = -1
          break
        }
      }
    }
  }
  return n - lost.length + count
}

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