[프로그래머스] 베스트앨범

문제

programmers.co.kr/learn/courses/30/lessons/42579

코딩테스트 연습 - 베스트앨범

문제해결방법

장르를 key, <고유번호, 재생횟수>를 저장하는 우선순위 큐를 value로 하는 HashMap과

장르를 key, 장르의 재생횟수를 value로 하는 HashMap 두 개를 사용해서 문제를 풀었다.

코드

package Programmers;
 
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.PriorityQueue;
 
class Music implements Comparable<Music> {
    String genre;
    int count;
    Music(String genre, int count) {
        this.genre = genre;
        this.count = count;
    }
 
    @Override
    public int compareTo(Music arg0) {
        return Integer.compare(arg0.count, this.count);
    }
}
 
public class Solution_베스트앨범 {
    
 
    public static int[] solution(String[] genres, int[] plays) {
        Map<String, PriorityQueue<int[]>> musicByGenres = new HashMap<>();
        Map<String, Integer> numberOfPlays = new HashMap<>();
        for(int i=0; i<genres.length; i++) {
            if(numberOfPlays.containsKey(genres[i])) {
                numberOfPlays.put(genres[i], numberOfPlays.get(genres[i])+plays[i]);
            } else numberOfPlays.put(genres[i], plays[i]);
 
            if(musicByGenres.containsKey(genres[i])) {
                musicByGenres.get(genres[i]).add(new int[]{i, plays[i]});
            } else {
                musicByGenres.put(genres[i], new PriorityQueue<int[]>(new Comparator<int[]>() {
                    @Override
                    public int compare(int[] arg0, int[] arg1) {
                        if(arg0[1]==arg1[1]) return Integer.compare(arg0[0], arg1[0]);
                        return Integer.compare(arg1[1], arg0[1]);
                    }
                }));
                musicByGenres.get(genres[i]).add(new int[]{i, plays[i]});
            }
        }
 
        PriorityQueue<Music> pq = new PriorityQueue<>();
        for(String key : numberOfPlays.keySet()) {
            pq.add(new Music(key, numberOfPlays.get(key)));
        }
        List<Integer> list = new ArrayList<>();
        while(!pq.isEmpty()) {
            Music cur = pq.poll();
            String genre = cur.genre;
            PriorityQueue<int[]> q = musicByGenres.get(genre);
            int i=0;
            while(!q.isEmpty() && i<2) {
                list.add(q.poll()[0]);
                i++;
            }
        }
        int[] answer = new int[list.size()];
        for(int i=0; i<answer.length; i++) answer[i] = list.get(i);
        return answer;
    }
    
    public static void main(String[] args) {
        String[] genres = {"classic", "pop", "classic", "classic", "pop"};
        int[] plays = {500, 600, 150, 800, 2500};
        System.out.println(Arrays.toString(solution(genres, plays)));
    }
}

코드에 대한 피드백이나 더 좋은 아이디어는 언제나 환영입니다.