난이도
레벨 3
문제
https://programmers.co.kr/learn/courses/30/lessons/42579
풀이
HashMap을 사용하는 문제로 HashMap에 관련된 다양한 연산을 적용하는 문제다.
문제 풀이 순서
[1] 정렬방법 1
1. hashmap에 장르별 play 횟수를 중첩해서 더해준다.
Map<String, Integer> map = new HashMap<>();
for(int i=0;i<genres.length;i++){
map.put(genres[i],map.getOrDefault(genres[i],0)+plays[i]);
} - hashmap의 key값만을 추출하여 리스트를 만들고, 리스트를 play 횟수를 기준으로 정렬
// key 값만 가져와서 genre에 넣어주기
ArrayList<String> genre = new ArrayList<>();
for(String key : map.keySet()){
genre.add(key);
}
// key 값에 해당하는 value를 내림차순으로 정렬
Collections.sort(genre,(o1,o2)->map.get(o2)-map.get(o1)); // **이해필요-
key 값을 정렬한 리스트에서 제일 많은 횟수를 재생한 장르부터 장르별 제일 많은 횟수가 플레이된 인덱스, 두번째로 많은 횟수가 플레이된 인덱스를 찾아 정답배열에 순서대로 넣어준다.
이때, 두번째로 많은 횟수가 플레이된 인덱스는 존재하지 않을 수도 있기 때문에 이를 잘 처리해준다.
ArrayList<Integer> list = new ArrayList<>();
for(int i=0;i<genre.size();i++){
String g = genre.get(i);
// 해당 장르의 음악중에서 play 횟수가 가장 큰 인덱스를 찾는다
int max = 0;
int firstIdx = -1;
for(int j=0;j<genres.length;j++){
if(g.equals(genres[j])&&max<plays[j]){
max=plays[j];
firstIdx =j;
}
}
// 해당 장르의 음악 중에서 play 횟수가 두번째로 큰 인덱스를 찾는다
max = 0;
int secondIdx = -1;
for(int j=0;j<genres.length;j++){
if(g.equals(genres[j]) && max < plays[j] && j != firstIdx){
max = plays[j];
secondIdx = j;
}
}
list.add(firstIdx);
if(secondIdx >=0) list.add(secondIdx); // secondIdx는 존재할 수도, 안할 수 도 있다
}[2] 정렬방법 2
1. hashmap에 장르별 play 횟수를 중첩해서 더해준다.
Map<String, Integer> map = new HashMap<>();
for(int i=0;i<genres.length;i++){
map.put(genres[i],map.getOrDefault(genres[i],0)+plays[i]);
} - hashmap을 value 기준(play 횟수)으로 정렬
ArrayList<Map.Entry<String, Integer>> mapList = new ArrayList<>(map.entrySet());
Collections.sort(mapList, new Comparator<>() {
@Override
public int compare(Map.Entry<String, Integer> o1, Map.Entry<String, Integer> o2) {
return o2.getValue().compareTo(o1.getValue());
}
});
-
key 값을 정렬한 리스트에서 제일 많은 횟수를 재생한 장르부터 장르별 제일 많은 횟수가 플레이된 인덱스, 두번째로 많은 횟수가 플레이된 인덱스를 찾아 정답배열에 순서대로 넣어준다.
이때, 두번째로 많은 횟수가 플레이된 인덱스는 존재하지 않을 수도 있기 때문에 이를 잘 처리해준다.
// 가장많은 플레이 횟수 재생 장르부터 첫번째 인덱스와 두번째 인덱스 고르고 정답배열에 넣기
ArrayList<Integer> list = new ArrayList<>();
for (Map.Entry<String, Integer> entry : mapList) {
// System.out.println(entry.getKey()+" "+ entry.getValue());
String genre = entry.getKey();
int max = 0;
int first = -1;
for (int i = 0; i < genres.length; i++) {
if (genre.equals(genres[i]) && max < plays[i]) {
max = plays[i];
first = i;
}
}
list.add(first);
max = 0;
int second = -1;
for (int i = 0; i < genres.length; i++) {
if (genre.equals(genres[i]) && max < plays[i] && first != i) {
max = plays[i];
second = i;
}
}
if (second >= 0) list.add(second);
}
- 정답리스트를 배열로 변환하여 반환한다.
int[] answer = new int[list.size()];
for (int i = 0; i < answer.length; i++) {
answer[i] = list.get(i);
}
return answer;코드
[1] 정렬 방법 1
import java.util.*;
class Solution {
public int[] solution(String[] genres, int[] plays) {
Map<String, Integer> map = new HashMap<>();
for(int i=0;i<genres.length;i++){
map.put(genres[i],map.getOrDefault(genres[i],0)+plays[i]);
}
// key 값만 가져와서 genre에 넣어주기
ArrayList<String> genre = new ArrayList<>();
for(String key : map.keySet()){
genre.add(key);
}
// key 값에 해당하는 value를 내림차순으로 정렬
Collections.sort(genre,(o1,o2)->map.get(o2)-map.get(o1));
ArrayList<Integer> list = new ArrayList<>();
for(int i=0;i<genre.size();i++){
String g = genre.get(i);
// 해당 장르의 음악중에서 play 횟수가 가장 큰 인덱스를 찾는다
int max = 0;
int firstIdx = -1;
for(int j=0;j<genres.length;j++){
if(g.equals(genres[j])&&max<plays[j]){
max=plays[j];
firstIdx =j;
}
}
// 해당 장르의 음악 중에서 play 횟수가 두번째로 큰 인덱스를 찾는다
max = 0;
int secondIdx = -1;
for(int j=0;j<genres.length;j++){
if(g.equals(genres[j]) && max < plays[j] && j != firstIdx){
max = plays[j];
secondIdx = j;
}
}
list.add(firstIdx);
if(secondIdx >=0) list.add(secondIdx); // secondIdx는 존재할 수도, 안할 수 도 있다
}
int[] result = new int[list.size()];
for(int i=0;i<list.size();i++){
result[i] = list.get(i);
}
return result;
}
}[2] 정렬 방법 2
import java.util.*;
class Solution {
public int[] solution(String[] genres, int[] plays) {
// 해시맵에 장르별 플레이 횟수 중첩해서 더해주기
HashMap<String, Integer> map = new HashMap<>();
for(int i=0;i<genres.length;i++){
map.put(genres[i],map.getOrDefault(genres[i],0)+plays[i]);
}
// 플레이 횟수기준을 정렬
ArrayList<Map.Entry<String,Integer>> mapList = new ArrayList<>(map.entrySet());
Collections.sort(mapList, new Comparator<>(){
@Override
public int compare(Map.Entry<String,Integer> o1, Map.Entry<String,Integer> o2){
return o2.getValue().compareTo(o1.getValue());
}
});
// 가장많은 플레이 횟수 재생 장르부터 첫번째 인덱스와 두번째 인덱스 고르고 정답배열에 넣기
ArrayList<Integer> list = new ArrayList<>();
for(Map.Entry<String,Integer> entry : mapList){
// System.out.println(entry.getKey()+" "+ entry.getValue());
String genre = entry.getKey();
int max =0;
int first = -1;
for(int i=0;i<genres.length;i++){
if(genre.equals(genres[i]) && max<plays[i]){
max = plays[i];
first=i;
}
}
list.add(first);
max =0;
int second = -1;
for(int i=0;i<genres.length;i++){
if(genre.equals(genres[i]) && max<plays[i] && first!=i){
max = plays[i];
second=i;
}
}
if(second>=0) list.add(second);
}
int[] answer = new int[list.size()];
for(int i=0;i<answer.length;i++){
answer[i] = list.get(i);
}
return answer;
}
}
Back-End Developer