[LV.3][JAVA] 베스트 앨범

https://school.programmers.co.kr/learn/challenges?tab=algorithm_practice_kit

프로그래머스

푸는 동안 사고과정 및 개선점

import java.util.*;

class Solution {
    public int[] solution(String[] genres, int[] plays) {
        ArrayList<Integer> arr = new ArrayList();
        HashMap<String, PriorityQueue<Song>> s = new HashMap();
        HashMap<String, Integer> g = new HashMap();
        for(int i = 0; i < genres.length; i++){
            g.put(genres[i], g.getOrDefault(genres[i], 0)+plays[i]); // 얘는 장르별 총 재생 수 넣게
            s.putIfAbsent(genres[i], new PriorityQueue<Song>((o1,o2)->{
                if(o2.plays == o1.plays){
                    // 재생수가 같으면 id가 낮은걸로 정렬
                    return o1.id - o2.id;
                }
                return o2.plays-o1.plays;
            }));
            s.get(genres[i]).add(new Song(i, plays[i]));
        } // 입력을 받았다. 각 장르 별로 plays 순 내림정렬, plays가 같으면 id 오름정렬.
        // 각 장르 별로 2개씩 뽑으면 되는데, 1개만 있다면 그거만 선택한다
        // 많이 재생된 장르를 먼저 수록한다 -> 이게 좀 애매
        
        // 키-값 쌍을 List로 변환
        List<Map.Entry<String, Integer>> list = new ArrayList(g.entrySet());
        // value를 기준으로 내림차순 정렬
        Collections.sort(list, (o1, o2) -> o2.getValue().compareTo(o1.getValue()));
        
        for(Map.Entry<String, Integer> entry: list){
            // 재생 수로 정렬된 키를 갖다가
            String genre = entry.getKey();
            int cnt = 0;
            while(cnt < 2){
                if(!s.get(genre).isEmpty()){
                    // 리스트가 남아있으면
                    arr.add(s.get(genre).poll().id);
                    cnt++;
                }else{
                    // 리스트가 비어있는 경우
                    break;
                }
            }
        }
        int[] answer = new int[arr.size()];
        int idx = 0;
        for(int id: arr){
            answer[idx++] = id;
        }
        
        return answer;
    }
}

class Song{
    int id; 
    int plays;
    Song(int id, int plays){
        this.id = id;
        this.plays = plays;
    }
}

IDE안쓰고 그냥 바로 풀었다. Map 조작에 조금 익숙해진 것 같다.

조건이 좀 많아서 까다로웠던 문제다. 

1. 총 재생 수가 많은 장르 순서대로
2. 장르 내에서 재생 수가 많은 순서대로
3. 재생 수가 같다면 id가 작은 순서대로 
4. 2개까지 뽑아서 id를 기록해라

작게 부분을 나눠서 작성하니까 조금 더 편했던 것 같다.

HashMap의 value로 Collection을 사용해보니까 활용할 수 있는 방향이 많을 것 같다고 생각했다. 

---

# 추가 공부

이번 문제 풀면서 검색헀던 건 HashMap에서 key 기준으로 정렬하는 방법이었다.

List<Map.Entry<String, Integer>> list = new ArrayList(g.entrySet());
Collections.sort(list, (o1, o2) -> o2.getValue().compareTo(o1.getValue()));

entrySet을 ArrayList에 넣을 수 있다는 걸 처음알았다.

entryset타입은 Map.Entry형태니까 주의해야되고, compareTo로 내림차순 정렬이 된 list를 만들었다

Collections.sort(list, (o1, o2) -> o1.getValue().compareTo(o2.getValue()));

이러면 오름차순이다.

compareTo의 앞에 있는걸 기준으로 compareTo안의 값이 크면 오름차순, 작으면 내림차순으로 정렬된다. Comparator를 람다로 작성한 형태라고 보면 될 것 같다.

 

도움이 됐다면 댓글이나 공감 버튼 한 번씩 누르고 가주세요!