[프로그래머스 | JavaScript] 코드 처리하기

문제 설명
문자열 code가 주어집니다.
code를 앞에서부터 읽으면서 만약 문자가 "1"이면 mode를 바꿉니다. mode에 따라 code를 읽어가면서 문자열 ret을 만들어냅니다.

mode는 0과 1이 있으며, idx를 0 부터 code의 길이 - 1 까지 1씩 키워나가면서 code[idx]의 값에 따라 다음과 같이 행동합니다.

mode가 0일 때
code[idx]가 "1"이 아니면 idx가 짝수일 때만 ret의 맨 뒤에 code[idx]를 추가합니다.
code[idx]가 "1"이면 mode를 0에서 1로 바꿉니다.
mode가 1일 때
code[idx]가 "1"이 아니면 idx가 홀수일 때만 ret의 맨 뒤에 code[idx]를 추가합니다.
code[idx]가 "1"이면 mode를 1에서 0으로 바꿉니다.
문자열 code를 통해 만들어진 문자열 ret를 return 하는 solution 함수를 완성해 주세요.

단, 시작할 때 mode는 0이며, return 하려는 ret가 만약 빈 문자열이라면 대신 "EMPTY"를 return 합니다.

 

문제 풀이

function solution(code) {
  let mode = 0;
  let ret = "";
  const codeArray = code.split("");

  codeArray.forEach((e, i) => {
    if (mode === 0) {
      e !== "1" && i % 2 === 0 ? (ret += e) : e === "1" ? (mode = 1) : null;
    } else {
      e !== "1" && i % 2 === 1 ? (ret += e) : e === "1" ? (mode = 0) : null;
    }
  });

  return ret === "" ? "EMPTY" : ret;

 

문제 설명에 나와있는 그대로 풀었다. 

mode가 0, 1인 경우를 먼저 구분하고, code[idx]가 1인지 아닌지, 1이라면 idx가 짝수인지 구분해서 풀었는데 여기서 시행착오가 한 가지 있었다. 

 

처음엔 요소만 인자로 받아서 forEach를 돌렸고 해당 codeArray의 요소인 e를 indexOf를 사용하여 idx를 계산했는데, 그렇게 하게 되면 1이 한 번만 나올 경우에는 정확히 계산하지만 1이 여러번 나올 경우 indexOf는 첫 번째 요소의 인덱스를 찾기 때문에 오류가 있었다. 

 

생각해보니 index는 forEach의 두 번째 인자로 넣으면 쉽게 해결될 문제였다.