🔎프로그래머스 - 베스트앨범

문제: https://school.programmers.co.kr/learn/courses/30/lessons/42579

카테고리: 해시

출처: 프로그래머스 코딩 테스트 고득점 Kit, https://school.programmers.co.kr/learn/challenges?tab=algorithm_practice_kit

---

✏️ 내 풀이

1. 각 장르별 총 재생 수를 저장하는 해시 테이블을 만든다.

   genreTotalPlay : { classic: 1450, pop: 3100}

2. 각 장르에 속한 노래와 인덱스를 저장하는 해시 테이블을 만든다.

   genreTable : {classic: [ [500, 0], [150, 2], [800, 3] ], pop: [ ~]}

3. 2에서 만든 해시 테이블에 속한 각 장르의 노래들을 재생수 내림차순으로 정렬하고, 최대 2개만 선택한다.

   genreTable : {classic: [ [ 800, 3 ], [ 500, 0 ] ], pop: [ [ 2500, 4 ], [ 600, 1 ] ]}

4. 총 재생 수가 높은 장르 순으로 장르 이름 정렬해서 새로운 배열 생성한다.

   keyArr : ["pop", "classic"] 

5. keyArr의 각 장르 이름을 key로 가지는 genreTable의 value 중 index만 이어붙인 새로운 배열을 생성한다.

   answer : [4,1,3,0]

---

🗒️ 내 코드

function solution(genres, plays) {

  // 각 장르별 총 재생 수를 저장하는 해시 테이블 genreTotalPlay :  { classic: 1450, pop: 3100}
  const genreTotalPlay = new Map();
  genres.forEach((genre, index) => {
    if (genreTotalPlay.get(genre)) {
      // table에 장르가 등록되어있다면
      const sum = genreTotalPlay.get(genre);
      genreTotalPlay.set(genre, sum + plays[index]);
    } else {
      genreTotalPlay.set(genre, plays[index]);
    }
  });

  // 각 장르에 속한 노래와 인덱스를 저장하는 해시 테이블 gerneTable :  {classic: [ [500, 0], [150, 2], [800, 3] ], pop: []}
  const genreTable = new Map();
  genres.forEach((gerne, index) => {
    if (genreTable.get(gerne)) {
      genreTable.set(gerne, [...genreTable.get(gerne), [plays[index], index]]);
    } else {
      genreTable.set(gerne, [[plays[index], index]]); // [500, 0] 이렇게 재생수와 index를 함께 저장 
    }
  });
  
  // 각 장르에 속한 노래들을 재생수 내림차순으로 정렬하고, 최대 2개만 선택하기
   // 'classic' => [ [ 800, 3 ], [ 500, 0 ] ],
  // 'pop' => [ [ 2500, 4 ], [ 600, 1 ] ]
  genreTable.forEach((arr, key) => {
    const newArr = arr.sort((a, b) => {
      return b[0] - a[0];
    });

    if (newArr.length <= 2) {
      genreTable.set(key, newArr);
    } else {
      genreTable.set(key, newArr.slice(0, 2));
    }
  });

  // 총 재생 수가 높은 장르 순으로 장르 이름 정렬
  const keyArr = [...genreTotalPlay.keys()].sort((a, b) => {
    return genreTotalPlay.get(b) - genreTotalPlay.get(a);
  });

  // 정렬된 장르 이름에 해당하는 노래들의 인덱스 이어 붙이기
  let answer = [];
  keyArr.forEach((key) => {
    genreTable.get(key).forEach((element) => answer.push(element[1]));
  });

  return answer;
}

solution(
  ["classic", "pop", "classic", "classic", "pop"],
  [500, 600, 150, 800, 2500]
);

---

✒️ Solution 및 개선방안

풀긴 했지만, 내 코드에는 뭔가 불필요한 로직이 있어 먼 길을 돌아 푼 것 같았고, 더 쉽고 간결하게 풀 수 있는 방법이 있을 것 같았다.

다음은 solution 코드이다.

function solution(genres, plays) {
  const genreMap = new Map();

  genres
    .map((genre, index) => {
      return [genre, plays[index]];
    }) // [classic, 500], [pop, 600] ...
    .forEach(([genre, play], index) => {
      // [classic, 500, 0] 비 구조화 할당

      // 기존 장르이름에 해당하는 데이터
      const data = genreMap.get(genre) || { total: 0, songs: [] }; // 장르가 처음 등록될 때는 undefined가 나올 것이므로, 이것 처리

      genreMap.set(genre, {
        total: data.total + play,
        songs: [...data.songs, { play, index }] // 기존의 songs가 [{500, 0}, {150,2} ...] 이렇게 있을 것이고, 여기다가 현재의 {play, index}를 추가해줄 것
          .sort((a, b) => b.play - a.play) // 재생 수 오름차순으로 정렬해서 2개만 짤라서 저장
          .slice(0, 2),
      });
      //  genremap : {
      //    classic: { total: 1450 , songs: [{ play: 800, index: 3 }, { play: 500, index: 0 } ] },
      //    pop : { total: 3100, songs: [{ play: 2500, index: 4 }, { play: 600, index: 1 } ] }
      //  }
      //
    });

  return [...genreMap.entries()] // [  [ 'classic', { total: 1450, songs: [Array] } ], [ 'pop', { total: 3100, songs: [Array] } ] ]
    .sort((a, b) => b[1].total - a[1].total) 
    .flatMap((item) => item[1].songs) // [ { play: 2500, index: 4 }, { play: 600, index: 1 }, { play: 800, index: 3 }, { play: 500, index: 0 } ]
    .map((song) => song.index);
}

• 처음 주어진 genres 배열과 plays를 조합해, 각 장르별 총 재생 수total과 노래 목록 songs를 포함한 해시테이블을 만들어 놓고 계산하는 것이 효율적으로 느껴졌다. 이를 만드는 과정에서 최대 재생 된 노래 2개만 선택하는 과정도 포함됐다.

  genremap : {
      classic: { total: 1450 , songs: [{ play: 800, index: 3 }, { play: 500, index: 0 } ] },
      pop : { total: 3100, songs: [{ play: 2500, index: 4 }, { play: 600, index: 1 } ] }
  }

• 논리연산자 || (or)을 이용해 data의 기본값을 설정해주는 기법도 잘 알아놔야겠다.

   const data = genreMap.get(genre) || { total: 0, songs: [] };

• flatMap() 함수를 이용해 각 요소들의 깊이를 -1씩 하여 새로운 배열을 만들어줄 수 있다.