[알고리즘] 프로그래머스 베트스앨범

처음풀어보는 프로그래머스 레벨 3 문제 였다.

문제부터 읽어보면 무슨 소리인지 몰랐는데,

가장 많이 재생된 장르를 골라 2개씩 고유번호(배열의 인덱스번호)를 결과 값에 담으면 된다.

 

이걸보면 pop는 총 3100번 > classic 는 1450번으로

pop부터 2개 담고 classic 2개를 담으면 된다.

import java.util.*;

class Solution {
    public Integer[] solution(String[] genres, int[] plays) {
        ArrayList<Integer> answerList = new ArrayList<Integer>();		
		HashMap<String, Integer> hashMap = new HashMap<String, Integer>();
		HashMap<Integer, String> genresMap = new HashMap<Integer, String>();
		HashMap<Integer, Integer> playsMap = new HashMap<Integer, Integer>();
		for (int i = 0; i < genres.length; i++) {
			hashMap.put(genres[i], hashMap.getOrDefault(genres[i], 0) + plays[i]);
			genresMap.put(i, genres[i]);
			playsMap.put(i, plays[i]);
		}
		List<String> keySetList = new ArrayList<>(hashMap.keySet());
		List<Integer> musicList = new ArrayList<>(genresMap.keySet());
		Collections.sort(keySetList, (o1, o2) -> (hashMap.get(o2).compareTo(hashMap.get(o1))));
		Collections.sort(musicList, (o1, o2) -> (playsMap.get(o2).compareTo(playsMap.get(o1))));

		for (String string : keySetList) {
			int count = 0;
			for (int i = 0; i < musicList.size(); i++) {
				if (genresMap.get(musicList.get(i)).equals(string)) {
					if (count == 2) {
						break;
					} else {
						count++;
						answerList.add(musicList.get(i));
					}
				}
			}
		}

		Integer[] answer = answerList.toArray(new Integer[0]);
		return answer;
    }
}

필자는 Map에 3개로 구현하고 정렬 후 답을 출력했다

Map<장르, 재생수>

Map<고유번호, 장르>

Map<고유번호, 장르>

이렇게 구현한 후 정렬후 비교과정을 거친후 결과 리스트에 담았다.