[Level 1] 체육복 - 프로그래머스 코딩테스트 연습

체육복 문제 풀이

문제

풀이

// javascript

function solution(n, lost, reserve) {
  // 수업이 가능한 학생 = 전체학생수 - 잃어버린 학생
  let answer = n - lost.length;

  for (let i = 0; i < lost.length; i++) {
    for (let j = 0; j < reserve.length; j++) {
      // 잃어버린 사람이 여분을 갖고 있는 경우
      if (lost[i] === reserve[j]) {
        // 각각 배열에서 제외
        lost.splice(i, 1);
        reserve.splice(j, 1);
        // 수업 가능한 학생 수 증가
        answer++;
        // 배열을 삭제해줬기 때문에 로직을 검증없이 넘어가는 index가 있을 수 있으므로 i를 빼줌
        i--;
        break;
      }
    }
  }

  for (let i = 0; i < lost.length; i++) {
    for (let j = 0; j < reserve.length; j++) {
      // 잃어버린 학생의 앞, 뒷번호의 학생이 여분을 갖고 있다면
      if (lost[i] - 1 === reserve[j] || lost[i] + 1 === reserve[j]) {
        // 각각 배열에서 제외
        lost.splice(i, 1);
        reserve.splice(j, 1);
        // 수업 가능한 학생 수 증가
        answer++;
        // 배열을 삭제해줬기 때문에 검증없이 넘어가는 index가 있을 수 있으므로 i를 빼줌
        i--;
        break;
      }
    }
  }

  return answer;
}

위 처럼 문제를 풀었는데, 다른 사람들의 풀이를 보고 반성하게 됐다. 다른 사람의 풀이 중 가장 짧은 풀이와, 내가 보기에 가장 이해가 쉬운 풀이를 첨부해본다. 참고하면서 공부를 많이 해야겠다.

다른 사람의 풀이

가장 짧은 코드

function solution(n, lost, reserve) {
  return (
    n -
    lost.filter((a) => {
      const b = reserve.find((r) => Math.abs(r - a) <= 1);
      if (!b) return true;
      reserve = reserve.filter((r) => r !== b);
    }).length
  );
}

내 기준 가장 이해가 쉬운 코드

function solution(n, lost, reserve) {
  var answer = new Array(n).fill(1);

  lost.forEach((val) => answer[val - 1]--);
  reserve.forEach((val) => answer[val - 1]++);

  for (let i = 0; i < answer.length; i++) {
    var clothLength = answer[i];

    if (clothLength === 2) {
      if (answer[i + 1] === 0 || answer[i - 1] === 0) {
        answer[i]--;
        answer[i + (answer[i + 1] === 0 ? 1 : -1)]++;
      }
    }
  }

  return answer.filter((x) => x >= 1).length;
}