[프로그래머스 C++] 베스트앨범

 

[프로그래머스 C++] 베스트앨범

 

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

프로그래머스

 

---

 

 

해결전략

 

map

자료구조

 

---

 

 

정답코드

 

#include <iostream>
#include <string>
#include <vector> 
#include <map> 
#include <algorithm> 
using namespace std; 

bool Compare(pair<string, int>& a, pair<string, int>& b) {
    return a.second > b.second; // 두 장르의 재생 횟수를 비교하여 내림차순으로 정렬
}

vector<int> solution(vector<string> genres, vector<int> plays) {
    
    map<string, int> myMap; // 장르별 총 재생 횟수를 저장할 맵
    map<string, vector<pair<int, int>>> playlistMap; // 장르별로 (재생 횟수, 인덱스) 쌍을 저장할 맵

    // genre별로 총 재생 횟수와 각 노래의 재생 횟수와 인덱스를 저장
    for (int i = 0; i < genres.size(); i++) {
        myMap[genres[i]] += plays[i]; // 장르별로 총 재생 횟수를 합산
        playlistMap[genres[i]].push_back({ plays[i], i }); // (재생 횟수, 인덱스) 쌍을 추가
    }

    // genre별 총 재생 횟수를 내림차순으로 정렬하기 위해 벡터로 변환
    vector<pair<string, int>> v(myMap.begin(), myMap.end());
    sort(v.begin(), v.end(), Compare); // 총 재생 횟수를 기준으로 내림차순 정렬

	vector<int> answer; // 정답을 저장할 벡터

    for (int i = 0; i < v.size(); i++) // 각 genre별로 상위 2개의 노래를 선택하여 정답에 추가
    {
        string genre = v[i].first; // 현재 장르
        if (playlistMap.find(genre) != playlistMap.end()) // 해당 genre가 존재하면
        { 
            sort(playlistMap[genre].begin(), playlistMap[genre].end(), [](pair<int, int> a, pair<int, int> b) {
                if (a.first == b.first) return a.second < b.second; // 재생 횟수가 같으면 인덱스 오름차순
                return a.first > b.first; // 재생 횟수가 다르면 내림차순
            });

            answer.push_back(playlistMap[genre][0].second); // 첫 번째 노래의 인덱스를 정답에 추가
            if (playlistMap[genre].size() >= 2) answer.push_back(playlistMap[genre][1].second); // 두 번째 노래의 인덱스를 정답에 추가 (있으면)
        }
    }

    return answer;
}

int main() {
    vector<string> testG1 = { "classic", "pop", "classic", "classic", "pop" };
    vector<int> testP1 = { 500, 600, 150, 800, 2500 };

    solution(testG1, testP1);

    return 0; 
}

 

 

---