[프로그래머스] 방금그곡 - 자바스크립트

문제

코딩테스트 연습 - [3차] 방금그곡

 

코드

function solution(m, musicinfos) {
  const arr = musicinfos.map((mi) => {
    const [start, end, title, code] = mi.split(",");
    const hour = end.slice(0, 2) - start.slice(0, 2);
    const minute = end.slice(3) - start.slice(3);
    const runtime = 60 * hour + minute;

    const codeArr = code.match(/[A-Z]#?/g);
    let stream = code.repeat(Math.floor(runtime / codeArr.length));
    stream += codeArr.slice(0,runtime % codeArr.length).join("");
    return [title, runtime, stream];
  });

  const answer = arr.filter(([_, __, stream]) => {
    let i = stream.indexOf(m);
    if (i === -1) return false;
    while (i !== -1) {
      if (stream[i + m.length] !== "#") return true;
      i = stream.indexOf(m, i + 1);
    }
  });
  if (!answer.length) return "(None)";

  answer.sort((a, b) => {
    if (a[1] === b[1]) return 0;
    return b[1] - a[1];
  });
  return answer[0][0];
}

 

풀이

코드 설명을 코드 라인에 주석으로 달아놨다.

 

문제 풀이 흐름은 다음과 같다.

 

musicinfos 배열의 각각의 문자열은 다음과 같이 구성된다.

"시작 시간, 끝난 시간, 음악 제목, 악보 정보"

 

각각의 문자열은 아래의 절차를 거친다.

 

1. (끝난 시간 - 시작 시간)으로 총 재생 시간을 구한다.
2. 악보 정보로부터 악보에 사용된 음들을 #을 포함해 [A, C#, C, D#] 등으로 나눠준다.
3. 재생 시간동안 재생된 음들을 연속적인 문자열로 기록한다.
4. [음악 제목, 총 재생 시간, 총 재생된 음 정보]의 배열로 반환한다.

 

위 절차를 거친 결과 배열을 가지고
각각의 결과에 대해 문자열 m이 포함되었는지 검사한다.

 

포함된 결과가 없다면 (None)을 반환한다.

 

총 재생 시간 순으로 내림차순 정렬한다.

총 재생 시간이 같을 때는 입력된 순서를 유지해준다.

 

이렇게 정렬된 결과 배열의 첫번째 요소의 제목이 정답이다.

 

주석을 포함한 코드

function solution(m, musicinfos) {
    const arr = musicinfos.map((mi)=>{ // 입력 배열의 문자열 각각에 대해 수행한다.
        const [start, end, title, code] = mi.split(","); // 문자열에서 ,를 구분자로 각 정보를 분리
        const hour = end.slice(0,2)-start.slice(0,2); // 문자열의 0~1번째 문자는 시간
        const minute = end.slice(3)-start.slice(3); // 문자열의 3~4번째 문자는 분
        const runtime = 60 * hour + minute; // 총 재생 시간

        const codeArr = code.match(/[A-Z]#?/g); // 알파벳, 알파벳#를 찾아 각각 배열의 인덱스에 분리한다.
        let stream = code.repeat(Math.floor(runtime/codeArr.length)); // 곡이 반복된 수 n만큼 악보 정보를 더한다.
        stream += codeArr.slice(0,runtime % codeArr.length).join(""); // 재생 도중에 멈춘 시점의 음까지를 추가로 더해준다.
        return [title, runtime, stream]; // 음악 제목, 총 재생 시간, 재생된 음 정보를 반환한다.
    })

    // 다음의 filter 함수는 조건을 충족하는(return true하는) 원소만 출력하는 함수
    const answer = arr.filter(([_,__,stream])=>{ // 각각의 결과 배열에 대해 다음을 수행한다.
        let i = stream.indexOf(m); // 재생된 음 정보에 문자열 m이 포함되었는지 검사한다.
        if(i === -1) return false; // 포함되어있지 않다면 넘어간다.
        while(i !== -1){ // 포함되어 있다면
            if(stream[i+m.length] !== "#") return true;
            i = stream.indexOf(m, i+1);
           // 찾은 문자 뒤에 #이 없다면 배열 정보를 그대로 반환한다.
           // ( stream = ABC#EFG 일 때 m= ABC를 찾을 수 있다.) 
           // 찾은 문자 뒤에 #이 있다면 정답이 아니므로, 찾은 문자 뒤에 m과 일치하는 문자가 있는지 찾아야 한다.
        }        
    });
    if(!answer.length) return "(None)"; // answer 배열에 원소가 없다면 조건에 일치하는 음악이 없는 것.

    answer.sort((a,b)=>{
        if(a[1] === b[1]) return 0;
        return b[1] - a[1];
    }) // 1번 인덱스(총 재생수)가 같다면 순서를 유지하고, 같지 않다면 내림차순 정렬한다.
    return answer[0][0]; // 입력 순서 + 내림차순을 반영했으므로 배열의 첫번째 원소의 제목이 정답이다.
}