체육복

체육복

// https://programmers.co.kr/learn/courses/30/lessons/42862
function solution(n, lost, reserve) {
  var answer = 0;

  let arr = [];
  for (let i = 0; i < n; i++) {
    arr[i] = 1;
  }

  for (let i = 0; i < reserve.length; i++) {
    arr[reserve[i] - 1]++;
  }

  for (let i = 0; i < lost.length; i++) {
    arr[lost[i] - 1]--;
  }

  for (let i = 0; i < arr.length; i++) {
    // 앞에사람꺼 빌리기
    if (arr[i] == 0) {
      if (arr[i - 1] == 2) {
        arr[i - 1]--;
        arr[i]++;
        continue;
      }
      // 뒤에사람꺼 빌리기
      else if (arr[i + 1] == 2) {
        arr[i + 1]--;
        arr[i]++;
        continue;
      }
    }
  }

  const result = arr.filter((value, index) => {
    return value != 0;
  });

  return result.length;
}
solution(5, [2, 4], [1, 3, 5]);

해설

• 체육복을 입을 수 있는 사람의 수를 구하는 문제다
• 체육복을 빌릴 수 있는데, 앞 뒤 사람의 체육복만 빌릴 수 있다
• 먼저 arr 배열을 초기화한다. lost는 -1, reserve는 +1
• 앞에서부터 순차적으로 앞사람확인하고 여벌이 있으면 빌린다
• 앞사람없으면 뒤에서 빌린다