[프로그래머스 Level 3] 베스트앨범 - JavaScript

🔗 문제 링크

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

프로그래머스

✏️ 풀이

function solution(genres, plays) {
    const genreMap = {};
    const genreCount = {};
    plays.forEach((time, num) => {
        const genre = genres[num];
        const highestCount = genreMap[genre];
        genreCount[genre] = (genreCount[genre] ?? 0) + time;
        if (!highestCount) {
            genreMap[genre] = [num];
            return;
        }
        if (highestCount[0] === undefined) {
            genreMap[genre][0] = num;
            return;
        }
        if (plays[highestCount[0]] < time) {
            genreMap[genre][1] = highestCount[0];
            genreMap[genre][0] = num;
            return;
        }
        if (highestCount[1] === undefined) {
            genreMap[genre][1] = num;
            return;
        }
        if (plays[highestCount[1]] < time) {
            genreMap[genre][1] = num;
        }
    });
    
    return Object
        .entries(genreCount)
        .sort((a, b) => b[1] - a[1])
        .reduce((acc, [genre, count]) => [...acc, ...genreMap[genre]], []);
}

plays를 순회하면서 각 노래가 해당 장르에서 재생 횟수가 1등 혹은 2등인지를 확인하여 그 여부를 갱신한다. genreMap에는 "장르: [1등 노래 번호, 2등 노래 번호]"의 형태로 값을 저장한다. 그리고 장르별 전체 재생횟수를 genreCount에 저장한다. plays의 순회가 끝나면 genreCount를 내림차순 정렬한 후, 해당 장르의 genreMap의 배열을 순서대로 이어붙여서 반환하면 된다.