[프로그래머스] 베스트앨범/Java - Lv.3

❓문제

프로그래머스

📌유형

해시

✍🏻풀이

장르별 총 횟수를 구한 HashMap을 구함.
pop에는 1, 4번이 classic에는 0, 2, 3 번으로 나눔으로써 각 장르별로 앨범을 나눔.
장르별 총 횟수를 내림차순으로 정렬하여 총 횟수가 가장 많은 장르 순으로 정렬함.
각 장르별로 앨범도 재생 횟수를 내림차순으로 정렬함.
앨범 중 재생횟수가 가장 많은 것을 answer에 추가. 앨범의 수가 2개 이상이라면 하나 더 추가.

💻코드

import java.util.*;
import java.io.*;

class Solution {
    public int[] solution(String[] genres, int[] plays) {
        Map<String, Integer> cnt = new HashMap<>();
        Map<String, HashMap<Integer, Integer>> album = new HashMap<>();
        
        for(int i = 0; i < genres.length; i++){
            if(!cnt.containsKey(genres[i])){
                HashMap<Integer, Integer> m = new HashMap<>();
                m.put(i, plays[i]);
                album.put(genres[i], m);
                cnt.put(genres[i], plays[i]);
            } else {
                album.get(genres[i]).put(i, plays[i]);
                cnt.put(genres[i], cnt.get(genres[i]) + plays[i]);
                
            }
        }
        
        List<String> keySet = new ArrayList<>(cnt.keySet());
        keySet.sort((o1, o2) -> cnt.get(o2) - (cnt.get(o1)));
        List<Integer> answer = new ArrayList();
        
        for(String i : keySet) {
            HashMap<Integer, Integer> map = album.get(i);
            List<Integer> key = new ArrayList(map.keySet());
            
            key.sort((o1, o2) -> map.get(o2) - (map.get(o1)));
            
            answer.add(key.get(0));
            if(key.size() > 1) // 현재 장르의 앨범 수가 2개 이상이라면 추가.
                answer.add(key.get(1));
        }
        
        return answer.stream().mapToInt(i -> i).toArray();
    }
}

💡새로 배운 내용

**List 람다함수 사용하여 정렬**
- 오름차순 정렬 List.sort((o1, o2) -> map.get(o1) - (map.get(o2)))
- 내림차순 정렬 List.sort((o1, o2) -> map.get(o2) - (map.get(o1)))

**스트림**
- 컬렉션, 배열 등에 저장된 요소들을 하나씩 참조하면서 코드를 실행할 수 있는 기능.
- 불필요한 for문을 사용하지 않고, 람다식을 활용하여 코드를 직관적이게 처리 가능.
**스트림 중간 연산**
- filter(Predicate<T> predicate): 주어진 조건에 맞는 요소만 선택.
- map(Function<T, R> mapper): 각 요소를 다른 형태로 변환.
- flatMap(Function<T, Stream<R>> mapper): 각 요소를 여러 요소로 확장하고 평탄화.
- distinct(): 중복된 요소를 제거.
- sorted(): 요소를 기본 정렬 순서로 정렬.
- peek(Consumer<T> action): 각 요소에 대해 주어진 동작을 수행하고, 동작 결과를 그대로 반환.
**스트림 최종 연산**
- forEach(Consumer<T> action): 각 요소에 대해 주어진 동작을 수행.
- toArray(): Stream의 요소를 배열로 변환..
- reduce(BinaryOperator<T> accumulator): 모든 요소를 하나의 값으로 축소
- collect(Collector<T, A, R> collector): Stream의 요소들을 컬렉션으로 수집.
- min(Comparator<T> comparator): Stream의 최솟값을 찾기.
- max(Comparator<T> comparator): Stream의 최댓값을 찾기.
- count(): Stream의 요소 개수를 반환.
- anyMatch(Predicate<T> predicate): 주어진 조건에 맞는 요소가 하나라도 있는지 확인.
- allMatch(Predicate<T> predicate): 모든 요소가 주어진 조건을 만족하는지 확인.
- noneMatch(Predicate<T> predicate): 모든 요소가 주어진 조건을 만족하지 않는지 확인.
- findFirst(): Stream의 첫 번째 요소를 반환.
- findAny(): Stream의 아무 요소나 반환.