[프로그래머스] 체육복 - 자바스크립트

문제 설명

 

문제 풀이

function solution(n, lost, reserve) {
  let answer = 0;
  let ans = [];
  for(let i = 0;i <= n; i++) {
      ans.push(1);
  }
  for(let i = 1;i <= n; i++) {
      for(let j = 0; j < lost.length; j++) {
          if(i === lost[j]) ans[i] = 0;
      }
      for(let j = 0; j < reserve.length; j++) {
          if(i === reserve[j]) ans[i] += 1;
      }
  }
  for(let i = 1; i <= n; i++){
      if(ans[i] === 0 && ans[i - 1] === 2) {
          ans[i - 1]  = 1;
          ans[i] = 1;
      }else if(ans[i] === 0 && ans[i + 1] === 2) {
          ans[i + 1] = 1;
          ans[i] = 1;
      }
  }
  for(let i = 1;i <= n; i++){
      if(ans[i] > 0) {
          answer++;
      }
  }
  return answer;
}

이중 for문을 돌려 해결하였다.

실행 결과

 

코드 채점하고 제출