[Java] 프로그래머스 - 베스트앨범 (+Comparable 사용)

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

프로그래머스

프로그래머스는 문제 유형이 뭔지 알려준다.

Hash 쓰는거길래 재밌을거 같아서 시도해보았다.

 

 

※ TIP

• Comparable을 꼭 이용하자!!!!!!

https://smile-development.tistory.com/92

Comparable 사용하기

 

 

인터페이스 없이 사용하려다가  해시 지옥을 맛봤다.. 오히려 돌아가는 일이니 그러지말자......

 

 

[코드]

import java.util.*;

class Solution {
    class Music implements Comparable<Music>{
        String genre;
        int index, play;
        
        public Music(String genre, int index, int play){
            this.genre = genre;
            this.index = index;
            this.play = play;
        }
        
        @Override
		public int compareTo(Music o) {
            
			if(this.play == o.play){
                return this.index - o.index;
            }
            
			return o.play - this.play;
		}
    }
    
    public int[] solution(String[] genres, int[] plays) {
        ArrayList<Integer> arr = new ArrayList<>();
        HashMap<String, Integer> sum = new HashMap<>();
        HashMap<String, PriorityQueue<Music>> list = new HashMap<>();
        HashSet<String> genreList = new HashSet<>();
        
        for(int i=0; i<genres.length; i++){
            String g = genres[i];
            int p = plays[i];
            
            // 총합 넣기
            sum.put(g, sum.getOrDefault(g,0)+p);
            
            // play 큰 수 -> index 작은 순
            PriorityQueue<Music> pq = list.getOrDefault(g, new PriorityQueue<>());
            pq.add(new Music(g, i, p));
            list.put(g, pq);
            
            genreList.add(g);
        }
        
        
        PriorityQueue<String> pq = new PriorityQueue<>((String s1, String s2)->(sum.get(s2)-sum.get(s1)));
        for(String g : genreList){
            pq.add(g);
        }
        
        while(!pq.isEmpty()){
            String g = pq.poll();
            PriorityQueue<Music> m = list.get(g);
            
            if(m.size()==1){
                arr.add(m.poll().index);
                continue;
            }
            
            arr.add(m.poll().index);
            arr.add(m.poll().index);
        }
        
        int[] ans = new int[arr.size()];
        for(int i=0; i<arr.size(); i++){
            ans[i] = arr.get(i);
        }
               
        return ans;
    }
}