import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;

class Solution {
    public int[] solution(String[] genres, int[] plays) {
        int[] answer = {};
        HashMap<String, Integer> genreMap = new HashMap<>(); // key : 장르명 / value : 총 재생횟수
        HashMap<String, ArrayList<Song>> songMap = new HashMap<>(); // key : 고유키 / value : Song객체 arrayList
        ArrayList<String> genreList = new ArrayList<>(); // genre명 담긴 list
        HashMap<Integer, Song> eachSongMap = new HashMap<>();
        for (int i = 0; i < genres.length; i++) {
            Song song = new Song(i, plays[i]);
            eachSongMap.put(i, song);
            ArrayList<Song> list = new ArrayList<>();
            if (songMap.get(genres[i]) == null) {
                songMap.put(genres[i], list);
                songMap.get(genres[i]).add(song);
            } else {
                songMap.get(genres[i]).add(song);
            }
            if (genreMap.get(genres[i]) == null) {
                genreList.add(genres[i]);
                genreMap.put(genres[i], plays[i]);
            } else {
                genreMap.replace(genres[i], genreMap.get(genres[i]) + plays[i]);
            }
        }
        int[] orderedList = new int[genreList.size()]; // 장르별 재생횟수 오름차순 정렬
        for (int i = 0; i < genreList.size(); i++) {
            orderedList[i] = genreMap.get(genreList.get(i));
        }
        Arrays.sort(orderedList);

        HashMap<Integer, String> reversedMap = new HashMap<>(); // key : 한 장르의 play 수 / value : 장르명
        for (int i = 0; i < genreList.size(); i++) {
            reversedMap.put(genreMap.get(genreList.get(i)), genreList.get(i));
        }
        ArrayList<Integer> finalResult = new ArrayList<>();
        int totalSize = 0;
        for (int i = orderedList.length - 1; i >= 0; i--) {
            ArrayList<Song> songList = songMap.get(reversedMap.get(orderedList[i]));
            int count = 0;
            int max = 0;
            if (songList.size() < 2) {
                max = 1;
            } else {
                max = 2;
            }
            int[] uniques = new int[2];
            uniques[1] = -1;
            int targetIndex = 0;
            boolean alreadyDelete = false;
            while (count < max) {
                int uniqueNumber = songList.get(0).uniqueNumber;
                int playTime = songList.get(0).playTime;
                for (int j = 1; j < songList.size(); j++) {
                    if (songList.get(j).playTime > playTime) {
                        playTime = songList.get(j).playTime;
                        uniqueNumber = songList.get(j).uniqueNumber;
                        targetIndex = j;
                    }
                }
                if (!alreadyDelete) {
                    songList.remove(targetIndex);
                    alreadyDelete = true;
                }
                uniques[count] = uniqueNumber;
                totalSize++;
                count++;
            }
            if (uniques[1] == -1) {
                finalResult.add(uniques[0]);
            } else {
                if (eachSongMap.get(uniques[0]) == eachSongMap.get(uniques[1])) {
                    if (uniques[0] < uniques[1]) {
                        finalResult.add(uniques[0]);
                        finalResult.add(uniques[1]);
                    } else {
                        finalResult.add(uniques[1]);
                        finalResult.add(uniques[0]);
                    }
                } else {
                    finalResult.add(uniques[0]);
                    finalResult.add(uniques[1]);
                }
            }
        }
        int index = 0;
        answer = new int[finalResult.size()];
        for (int i = 0; i < finalResult.size(); i++) {
            answer[index] = finalResult.get(i);
            index++;
        }
        return answer;
    }
}


class Song { //멤버 변수 : 고유번호 + 재생횟수
    int uniqueNumber;
    int playTime;

    public Song(int uniqueNumber, int playTime) {
        this.uniqueNumber = uniqueNumber;
        this.playTime = playTime;
    }
}