[Algorithm] 프로그래머스 - 베스트 앨범.java

1. 문제

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

프로그래머스

 

 

2. 풀이

1. 장르 재생횟수 많은 순서

2. 노래 재생횟수 많은 순서

3. 고유번호 작은 순서

map 2개를 두고

- 노래 정보 map (해시맵의 해시맵)

- 재생횟수 map

 

장르의 key값만 뽑은 리스트를 정렬할 때, 정렬 조건에 장르 재생횟수가 많은 순 조건 추가

노래의 key값만 뽑은 리스트를 정렬할 때, 정렬 조건에 노래 재생횟수가 많은 순 조건 추가

answer 리스트에 개수 되는대로 add (단, 2개까지)

 

유의할 점은

List<Integer>를 int[] 로 맵핑하는 방법인데,

stream과 mapToInt를 사용하였다.

 

1. mapToInt 원리

2. 해시맵에서 value가 map 형태면 put했을 때 이미 key가 있어도 value 업데이트가 아니라 중복되는 key에 새로운 value가 추가되는건가? value 안의 key도 체크하는건가?

 

3. 코드

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

class Solution {
    public int[] solution(String[] genres, int[] plays) {
        List<Integer> answer = new ArrayList<> ();
		Map<String, Integer> countMap = new HashMap<> ();
		Map<String, Map<Integer, Integer>> musicMap = new HashMap <> ();
		
		for (int i=0; i<genres.length; i++) {
			if (countMap.containsKey(genres[i])) {
				countMap.put(genres[i], countMap.get(genres[i]) + plays[i]);
				musicMap.get(genres[i]).put(i, plays[i]);
			}
			else {
				Map<Integer, Integer> map = new HashMap<> ();
				map.put(i, plays[i]);
				countMap.put(genres[i], plays[i]);
				musicMap.put(genres[i], map);
			}
		}
		List<String> genreList = new ArrayList<> (countMap.keySet());
		Collections.sort(genreList, (s1, s2) -> countMap.get(s2) - countMap.get(s1));
		for (String genre : genreList) {
			Map<Integer, Integer> music = (HashMap<Integer, Integer>) musicMap.get(genre);
			List<Integer> musicList = new ArrayList<> (music.keySet());
			Collections.sort(musicList, (s1, s2) -> music.get(s2) - music.get(s1));
			
			answer.add(musicList.get(0));
			if (musicList.size() > 1) {
				answer.add(musicList.get(1));
			}
		}
		
		return answer.stream()
				.mapToInt(i->i)
				.toArray();
    }
}

4. 사담