베스트앨범

[fkdl0048]

#include <string>
#include <vector>
#include <unordered_map>
#include <algorithm>
#include <iostream>

using namespace std;

struct Song {
    int id;
    int plays;
};

struct Genre {
    int totalPlays;
    vector<Song> songs;
};

vector<int> solution(vector<string> genres, vector<int> plays) {
    unordered_map<string, Genre> genreMap;
    
    for (int i = 0; i < genres.size(); i++) {
        genreMap[genres[i]].totalPlays += plays[i];
        genreMap[genres[i]].songs.push_back({i, plays[i]});
    }
    
    vector<pair<string, Genre>> genreList(genreMap.begin(), genreMap.end());
    sort(genreList.begin(), genreList.end(), [](const auto& a, const auto& b) {
        return a.second.totalPlays > b.second.totalPlays;
    });
    
    vector<int> result;
    
    for (auto& genre : genreList) {
        vector<Song>& songs = genre.second.songs;
        sort(songs.begin(), songs.end(), [](const Song& a, const Song& b) {
           if (a.plays == b.plays) {
               return a.id < b.id;
           }
            return a.plays > b.plays;
        });
        
        for (size_t i = 0; i < min(songs.size(), static_cast<size_t>(2)); i++) {
            result.push_back(songs[i].id);
        }
    }
    
    return result;
}

• 문제를 읽고 조건에 맞게 설정하는 것이 중요, 명확하게 풀이하려면 따로 구조체를 만들어서 사용자 정의로 정렬하는 것이 좋아 보임
• min의 경우 둘 중에 하나의 자료형에 맞게 캐스팅하는 것이 좋을 듯